UVA12879 Golf Bot

UVA12879 Golf Bot

或许我应该把这道题放到I.BZOJ3513: [MUTC2013]idiots前面的QaQ。

这题完全就是那题的子问题，直接暴力FFT一下即可。连特殊情况都不需要考虑。

感觉这题被恶评了，应该是紫-，不可能到黑。

代码：

#include<bits/stdc++.h>
using namespace std;
const int lg=19,lim=(1<<lg);
const double pi=acos(-1);
int n,m,cnt[lim+5],res,rev[lim+5];
struct cp{
	double x,y;
	cp(double u=0,double v=0){x=u,y=v;}
	friend cp operator +(const cp &u,const cp &v){return cp(u.x+v.x,u.y+v.y);}
	friend cp operator -(const cp &u,const cp &v){return cp(u.x-v.x,u.y-v.y);}
	friend cp operator *(const cp &u,const cp &v){return cp(u.x*v.x-u.y*v.y,u.x*v.y+u.y*v.x);}
}f[lim+5];
void FFT(cp *a,int tp){
	for(int i=0;i<lim;i++)if(i<rev[i])swap(a[i],a[rev[i]]);
	for(int md=1;md<lim;md<<=1){
		cp rt=cp(cos(pi/md),tp*sin(pi/md));
		for(int stp=md<<1,pos=0;pos<lim;pos+=stp){
			cp w=cp(1,0);
			for(int i=0;i<md;i++,w=w*rt){
				cp x=a[pos+i],y=w*a[pos+md+i];
				a[pos+i]=x+y;
				a[pos+md+i]=x-y;
			}
		}
	}
}
int main(){
	for(int i=0;i<lim;i++)rev[i]=(rev[i>>1]>>1)|((i&1)<<(lg-1));
	while(scanf("%d",&n)!=EOF){
		for(int i=0;i<lim;i++)f[i]=cp(0,0),cnt[i]=0;
		f[0].x=1;
		res=0;
		for(int i=0,x;i<n;i++)scanf("%d",&x),f[x].x+=1;
		FFT(f,1);
		for(int i=0;i<lim;i++)f[i]=f[i]*f[i];
		FFT(f,-1);
		for(int i=0;i<lim;i++)cnt[i]=(int)(f[i].x/lim+0.5);
//		for(int i=1;i<=10;i++)printf("%d ",cnt[i]);
		scanf("%d",&m);
		for(int i=0,x;i<m;i++)scanf("%d",&x),res+=(cnt[x]!=0);
		printf("%d
",res);
	}
	return 0;
}