给出
n
≤
2
e
3
n\leq2e3
n≤2e3个点,以及
q
≤
2
e
3
q\leq2e3
q≤2e3个点,保证所有的点都互不相同,对于每次询问的这个点,求问有多少对
(
u
,
v
)
(u,v)
(u,v),和当前这个点构成一个直角三角形。
分成两种情形:
1.询问的点是直角的顶点:对n个点以询问点为中心极角排序。然后枚举每个点,二分出逆时针90°的点有多少个,只考虑一个方向就不会重复计算。
2.询问的点是不是直角的顶点:把q个询问点离线,然后枚举n个点中的每个点作为中心,把剩下来的n-1个点极角排序,枚举每个询问点,求出方案加到询问点的答案上。
#include <bits/stdc++.h>
using namespace std;
typedef long long LL;
const int N=4e3+7;
struct Point {
int id,x,y;
Point() {}
Point(int x,int y) {
this->x=x;
this->y=y;
}
}p[N],t[N],o;
Point operator -(const Point &a,const Point &b) {
return Point(a.x-b.x,a.y-b.y);
}
bool operator ^(const Point &a,const Point &b) {
return (1ll*a.x*b.y-1ll*a.y*b.x)>0;
}
int Quadrant(const Point &a) {
if(a.x>0&&a.y>=0) return 1;
else if(a.x<=0&&a.y>0) return 2;
else if(a.x<0&&a.y<=0) return 3;
else if(a.x>=0&&a.y<0) return 4;
}
bool operator <(const Point &c,const Point &d) {
Point a=c-o,b=d-o;
int qa=Quadrant(a);
int qb=Quadrant(b);
if(qa!=qb) return qa<qb;
return a^b;
}
int ans[N];
int main() {
int n,q;
scanf("%d%d",&n,&q);
for(int i=1;i<=n;i++) {
scanf("%d%d",&p[i].x,&p[i].y);
p[i].id=i;t[i]=p[i];
}
for(int i=n+1;i<=n+q;i++) {
scanf("%d%d",&p[i].x,&p[i].y);
p[i].id=i;
o=p[i];
sort(p+1,p+1+n);
for(int j=1;j<=n;j++) {
int x=p[j].x-o.x;
int y=p[j].y-o.y;
Point a=Point(o.x-y,o.y+x);
ans[i]+=upper_bound(p+1,p+1+n,a)-lower_bound(p+1,p+1+n,a);
}
}
for(int i=1;i<=n;i++) {
memcpy(p+1,t+1,sizeof(Point)*n);
o=p[i];
swap(p[i],p[n]);
sort(p+1,p+n);
for(int j=n+1;j<=n+q;j++) {
int x=p[j].x-o.x;
int y=p[j].y-o.y;
Point a=Point(o.x-y,o.y+x);
Point b=Point(o.x+y,o.y-x);
ans[p[j].id]+=upper_bound(p+1,p+n,a)-lower_bound(p+1,p+n,a);
ans[p[j].id]+=upper_bound(p+1,p+n,b)-lower_bound(p+1,p+n,b);
}
}
for(int i=n+1;i<=n+q;i++) {
printf("%d\n",ans[i]);
}
return 0;
}