背景:
考场上30分,后来听说是FFT,觉得好高级,感觉这种问题好神仙。
改用
M
a
r
k
d
o
w
n
Markdown
Markdown。
题意:
在平面直角坐标系给
n
n
n个点,每个点有一个对应的类型(
012
012
012),求两个相同类型夹中点为不同类型的点的方案数。
思路:
由数学知识可以知道,对于三个点
(
x
1
,
y
1
)
(x_1,y_1)
(x1,y1),
(
x
2
,
y
2
)
(x_2,y_2)
(x2,y2),
(
x
3
,
y
3
)
(x_3,y_3)
(x3,y3),当
(
x
2
,
y
2
)
(x_2,y_2)
(x2,y2)为另外两点中点时,必然有
x
2
=
x
1
+
x
3
2
,
y
2
=
y
1
+
y
3
2
x_2=\dfrac{x_1+x_3}{2},y_2=\dfrac{y_1+y_3}{2}
x2=2x1+x3,y2=2y1+y3,同理,知道后者也可推出前者。
考虑将二维问题转化为一维问题。
对于每一个点的坐标
(
x
i
,
y
i
)
(x_i,y_i)
(xi,yi)变为
1000
∗
x
i
+
y
i
1000*x_i+y_i
1000∗xi+yi。那么对于三个点
1000
∗
x
1
+
y
1
1000*x_1+y_1
1000∗x1+y1,
1000
∗
x
2
+
y
2
1000*x_2+y_2
1000∗x2+y2,
1000
∗
x
3
+
y
3
1000*x_3+y_3
1000∗x3+y3,若
1000
∗
x
2
+
y
2
1000*x_2+y_2
1000∗x2+y2为其它两点的中点,必然有
1000
∗
x
2
+
y
2
=
1000
∗
x
1
+
y
1
+
1000
∗
x
3
+
y
3
2
1000*x_2+y_2=\dfrac{1000*x_1+y_1+1000*x_3+y_3}{2}
1000∗x2+y2=21000∗x1+y1+1000∗x3+y3,即
(
1000
∗
x
2
+
y
2
)
∗
2
=
1000
∗
x
1
+
y
1
+
1000
∗
x
3
+
y
3
(1000*x_2+y_2)*2=1000*x_1+y_1+1000*x_3+y_3
(1000∗x2+y2)∗2=1000∗x1+y1+1000∗x3+y3。也就是说当前仅当满足
y
∗
2
=
x
+
z
y*2=x+z
y∗2=x+z时,
y
y
y为
x
,
z
x,z
x,z的中点。同理,知道后者也可推出前者。
问题就变得简单了。
考虑设
y
y
y的类型为
0
0
0,其它类型同理。
构造
A
i
=
[
t
y
p
e
i
=
0
]
A_i=[type_i=0]
Ai=[typei=0],然后另
S
i
=
A
i
∗
A
i
S_i=A_i*A_i
Si=Ai∗Ai。我们有上面可以知道,若
y
y
y点为
x
,
z
x,z
x,z中点,其结果必然是偶数,而
x
,
z
x,z
x,z,
z
,
x
z,x
z,x为一种情况,因此结果需要除以
2
2
2。所以由
S
i
S_i
Si来计算结果即可。
时间复杂度
Θ
(
n
l
o
g
n
)
\Theta(nlogn)
Θ(nlogn)。
代码:
// luogu-judger-enable-o2
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cmath>
using namespace std;
const double pai=acos(-1.0);
struct comp
{
double x,y;
comp(double xx=0,double yy=0):x(xx),y(yy) {}
friend comp operator+(const comp &x,const comp &y) {return comp(x.x+y.x,x.y+y.y);}
friend comp operator-(const comp &x,const comp &y) {return comp(x.x-y.x,x.y-y.y);}
friend comp operator*(const comp &a,const comp &b) {return comp(a.x*b.x-a.y*b.y,a.x*b.y+b.x*a.y);}
}a[2500000];
int limit=1,n,l=0,ma=0;
int r[2500000],ans[2500000],id[2500000];
int type[2500000];
void init(int ma)
{
while(limit<=(ma<<1))
limit<<=1,l++;
for(int i=1;i<=limit;i++)
r[i]=((r[i>>1]>>1)|((i&1)<<(l-1)));
}
void FFT(comp *now,int ty)
{
for(int i=0;i<limit;i++)
if(i<r[i]) swap(now[i],now[r[i]]);
for(int mid=1;mid<limit;mid<<=1)
{
comp wn(cos(pai/mid),ty*sin(pai/mid));
for(int j=0,R=(mid<<1);j<limit;j+=R)
{
comp w(1,0);
for(int k=0;k<mid;k++,w=w*wn)
{
comp x=now[j+k],y=w*now[j+k+mid];
now[j+k]=x+y;
now[j+k+mid]=x-y;
}
}
}
}
void work(int x)
{
for(int i=0;i<=limit;i++)
a[i]=(comp){0,0};
for(int i=0;i<=ma;i++)
if(type[i]==x) a[i].x=1; else a[i].x=0;
FFT(a,1);
for(int i=0;i<=limit;i++)
a[i]=a[i]*a[i];
FFT(a,-1);
for(int i=0;i<=(ma<<1);i+=2)
if(type[i/2]&&type[i/2]!=x) ans[i/2]+=(int)floor(a[i].x/limit+0.001)/2;
}
int main()
{
int x,y,z;
scanf("%d",&n);
for(int i=1;i<=n;i++)
{
scanf("%d %d %d",&x,&y,&z);
type[x*1000+y]=z+1;
id[i]=x*1000+y;
ma=max(ma,x*1000+y);
}
init(ma);
work(1),work(2),work(3);
for(int i=1;i<=n;i++)
printf("%d\n",ans[id[i]]);
}