JZOJ 3129. 【WinterCamp 2013】数三角形

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)，那么她称这个三角形为“黄金三角形”。原点不会落在任何一对奶牛的连线上。另外，不会有奶牛在原点。

给出奶牛的坐标，计算出有多少个“黄金三角形”。

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

Hint

考虑五只牛，坐标分别为(-5,0), (0,2), (11,2), (-11,-6), (11,-5)。

下图是由贝西视角所绘出的图示。

      ............|............
      ............*..........*.
      ............|............
      -------*----+------------
      ............|............
      ............|............
      ............|............
      ............|............
      ............|..........*.
      .*..........|............
      ............|............

Solution

• 正难则反，总组合数为 $C_n^3$ ，只要求出不过原点的三角形个数即可。

• 现将每个点进行极角排序（以原点为顶点、两边分别为 x的正半轴 和 原点与这个点的连边）

• 按角从小到大枚举，统计连续的一段，使得组成的三角形不包含原点，则该段可以两两匹配。

• 判断其是否包含原点，只需要判断两向量的叉积是否大于0即可（大于0则不包含）。

• 时间复杂度为 $O(N\ log\ N)$ 。

Code

#include<cstdio>
#include<algorithm>
#include<cctype>
#include<complex>
using namespace std;
typedef long long LL;
typedef complex<LL> point;
const int N=1e5+2;
point a[N<<1];
inline int read()
{
    int X=0,w=0; char ch=0;
    while(!isdigit(ch)) w|=ch=='-',ch=getchar();
    while(isdigit(ch)) X=(X<<3)+(X<<1)+(ch^48),ch=getchar();
    return w?-X:X;
}
inline bool upper(point x)
{
    return x.imag()>0 || !x.imag() && x.real()>0;
}
inline LL cross(point x,point y)
{
    return x.real()*y.imag()-x.imag()*y.real();
}
inline bool cmp(point x,point y)
{
    bool xx=upper(x),yy=upper(y);
    if(xx && !yy) return true;
    if(!xx && yy) return false;
    return cross(x,y)>0;
}
int main()
{
    int n=read();
    for(int i=1;i<=n;i++) a[i].real()=read(),a[i].imag()=read();
    sort(a+1,a+1+n,cmp);
    LL ans=n*1LL*(n-1)*(n-2)/6;
    for(int i=1;i<=n;i++) a[i+n]=a[i];
    for(int i=1,j=1;i<=n;i++)
    {
        while(j<i+n-1 && cross(a[i],a[j+1])>0) j++;
        ans-=(j-i)*1LL*(j-i-1)/2;
    }
    printf("%lld",ans);
    return 0;
}