HDU-6731 Angle Beats（极角排序）

题目： http://acm.hdu.edu.cn/showproblem.php?pid=6731

题意：给定平面n个点坐标，q次询问，每次给出一个询问点，求由此点和n中任意两点能组成多少个直角三角形。

思路：

    先考虑把给定点看作直角顶点，看能组成多少直角三角形。以给定点为中心，对其他n个点进行极角排序（先按象限，再按极角从小到大），由于已经排序好，每次要找一个向量的垂直向量二分即可（O(nlg(n))），当然也可以循环遍历一遍查找O(n)，不过写起来比较麻烦。
    第二种的情况是给定点不作为直角顶点，此时构成直角三角形的直角顶点为给定的n个点中的某个点，且另外两点分别为n个点中的另一点和要询问的给定点，考虑离线，把n中的每个点当作直角点，看找到构成直角三角形的剩余两个点是否一个是n个点中的另一个且另一个点为询问点中的一个，原理同第一种情况相同。

代码：

#include <bits/stdc++.h>
#define LL long long
using namespace std;
const int maxn = 2e3+3;

struct Point{
    LL x, y, id;
    Point(){}
    Point(LL _x, LL _y){x = _x, y = _y;}
    LL operator ^ (const Point & p) const {return x*p.y - y*p.x;}
    LL operator * (const Point & p) const {return x*p.x + y*p.y;}
    Point operator - (const Point & p) const {return Point(x-p.x, y-p.y);}
    bool operator < (const Point & p) const {return x==p.x?y<p.y:x<p.x;}
};
LL dot(Point a, Point b, Point c){return (b-a)*(c-a);}
LL cross(Point a, Point b, Point c){return (b-a)^(c-a);}

int Quadrant(Point a){
    if(a.x>0 && a.y>=0) return 1;
    if(a.x<=0 && a.y>0) return 2;
    if(a.x<0 && a.y<=0) return 3;
    if(a.x>=0 && a.y<0) return 4;
}
bool cmp(Point a, Point b){
    if(Quadrant(a) == Quadrant(b)) return (a^b)>0;
    return Quadrant(a) < Quadrant(b);
}
Point p[maxn], q[maxn], t[maxn*2];
int n, m, ans[maxn];
int main()
{
    while(~scanf("%d%d", &n, &m)){
        for(int i=1; i<=n; i++) scanf("%lld%lld", &p[i].x, &p[i].y);

        memset(ans, 0, sizeof ans);
        for(int i=1; i<=m; i++){
            scanf("%lld%lld", &q[i].x, &q[i].y);
            memset(t, 0, sizeof t);
            for(int j=1; j<=n; j++) t[j] = p[j]-q[i];
            sort(t+1, t+n+1, cmp);

            for(int j=1; j<=n; j++){
                Point v(-t[j].y, t[j].x);
                int k = lower_bound(t+1, t+n+1, v, cmp)-t;
                while(t[j]*t[k] == 0 && (t[j]^t[k])>0) k = k+1>n?1:k+1, ans[i] ++;
            }
        }
        for(int i=1; i<=n; i++){
            int cnt = 0; memset(t, 0, sizeof t);
            for(int j=1; j<=n; j++) if(i!=j) t[++cnt] = p[j]-p[i], t[cnt].id = 0;
            for(int j=1; j<=m; j++) t[++cnt] = q[j]-p[i], t[cnt].id = j;
            sort(t+1, t+cnt+1, cmp);

            for(int j=1; j<=cnt; j++){
                Point v(-t[j].y, t[j].x);
                int k = lower_bound(t+1, t+cnt+1, v, cmp) - t;
                while(t[j]*t[k]==0 && (t[j]^t[k])>0) {
                    if(t[j].id != 0 && t[k].id == 0) ans[t[j].id] ++;
                    else if(t[j].id == 0 && t[k].id != 0) ans[t[k].id] ++;
                    k = k+1>cnt?1:k+1;
                }
            }
        }
        
        for(int i=1; i<=m; i++)
            printf("%d\n", ans[i]);
    }
}
/*
7 1
3 0
0 -3
-3 0
-2 -2
-2 1
-1 3
3 -3

0 0 //6
*/