Luogu2992[USACO10OPEN] Triangle Counting

洛谷链接：https://www.luogu.org/problemnew/show/P2992
bzoj链接：http://www.lydsy.com/JudgeOnline/problem.php?id=1914

Triangle Counting

Description

在一只大灰狼偷偷潜入Farmer Don的牛群被群牛发现后，贝西现在不得不履行着她站岗的职责。从她的守卫塔向下瞭望简直就是一件烦透了的事情。她决定做一些开发智力的小练习，防止她睡着了。想象牧场是一个X，Y平面的网格。她将N只奶牛标记为1…N (1 <= N <= 100,000)，每只奶牛的坐标为X_i,Y_i (-100,000 <= X_i <= 100,000;-100,000 <= Y_i <= 100,000; 1 <= i <=N)。然后她脑海里想象着所有可能由奶牛构成的三角形。如果一个三角形完全包含了原点(0,0)，那么她称这个三角形为“黄金三角形”。原点不会落在任何一对奶牛的连线上。另外，不会有奶牛在原点。给出奶牛的坐标，计算出有多少个“黄金三角形”。顺便解释一下样例，考虑五只牛，坐标分别为(-5,0), (0,2), (11,2), (-11,-6), (11,-5)。下图是由贝西视角所绘出的图示。

Input

第一行:一个整数: N
第2到第N+1行: 每行两个整数X_i，Y_i，表示每只牛的坐标

Output

第一行: 一行包括一个整数，表示“黄金三角形的数量”

Sample Input

5
-5 0
0 2
11 2
-11 -6
11 -5

Sample Output

5

题解

每个点和原点的交点所在的直线能将平面分成两部分，显然，在其中任意一部分中的任意两个点都无法与原来的点连成“黄金三角形”。我们只需要枚举每个点，统计其一边的点的数量n，那么n×（n-1）就是“非黄金三角形”的数量，而跨过这条直线形成的“黄金三角形”必定会被其他点统计到（大家画图+感性理解一下），所以“黄金三角形”数量=三角形数量（n×（n-1）×（n-2）÷6）-“非黄金三角形”数量，我们就得出了答案。  
然而事情还没有想象的那么简单，我们如果每个点暴力统计的话O(n^2)依然T的飞起。我们注意到，在直线一侧的点其实是可以二分的，我们将所有点按与原点连接的直线的斜率排序，统计就可以二分查找在直线一侧的点。这样复杂度就是O(3nlogn)的了，已经可以A掉这道题。  
然而其实还有更优的解法，在统计直线一侧的点的时候，其实信息是可以复用的，如果我们按斜率从小到大查找的话，那么查找的点很大一部分是重复的，我们复用这部分信息，每次只需要查找一小部分的点，总共复杂度就是O(n+nlogn)。

代码

#include<bits/stdc++.h>
using namespace std;
int n;
struct pt{long long x,y;double ang;};
long long operator * (pt a,pt b){return a.x*b.y-a.y*b.x;}
bool operator < (const pt &a,const pt &b){return a.ang<b.ang;}
pt a[100005];
void in()
{
    scanf("%d",&n);
    for(int i=1;i<=n;++i)
    scanf("%lld%lld",&a[i].x,&a[i].y),a[i].ang=atan2(a[i].y,a[i].x);
}
void ac()
{
    sort(a+1,a+n+1);
    long long cot=0,ans=0,r=2;
    for(int i=1;i<=n;++i)
    {
        while(r!=i&&a[i]*a[r]>=0) cot++,r=r%n+1;
        ans+=1ll*cot*(cot-1)/2;
        cot--;
    }
    printf("%lld",1ll*n*(n-1)*(n-2)/6-ans);
}
int main()
{
    in();ac();
    return 0;
}