题目: 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
*/