FFT--UVA12633 Super Rooks on Chessboard

传送门

不那么裸的$FFT$，题目给的标号方式不太好，可以考虑改成从上到下为$1-n$，从右到左为$1-m$，这样从左上到右下的一条斜线上的点的横纵坐标的和是一样的，就可以用$FFT$来解决，斜着的只要记一下就行了，横和竖都可以用斜着的那种方法来计算emmm这样说可能不太清楚一看代码就懂了

#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cstring>
#include<cmath>
#define maxn 200005
#define LL long long
using namespace std;
const double Pi=acos(-1.0);

inline int rd(){
	int x=0,f=1;char c=' ';
	while(c<'0' || c>'9') f=c=='0'?-1:1,c=getchar();
	while(c<='9' && c>='0') x=x*10+c-'0',c=getchar();
	return x*f;
}

int t,n,m,cnt,row[maxn],line[maxn],qwq[maxn],limit=1,l,rev[maxn];

struct complex{
	double x,y;
	complex(double xx=0,double yy=0) {x=xx,y=yy;}
}a[maxn],b[maxn];
complex operator +(complex a,complex b){return complex(a.x+b.x,a.y+b.y);}
complex operator -(complex a,complex b){return complex(a.x-b.x,a.y-b.y);}
complex operator *(complex a,complex b){return complex(a.x*b.x-a.y*b.y,a.x*b.y+a.y*b.x);}

inline void FFT(complex *F,int type){
	for(int i=0;i<limit;i++) if(i<rev[i]) swap(F[i],F[rev[i]]);
	for(int mid=1;mid<limit;mid<<=1){
		complex Wn(cos(Pi/mid),1.0*type*sin(Pi/mid));
		for(int r=mid<<1,j=0;j<limit;j+=r){
			complex w(1,0);
			for(int k=0;k<mid;k++,w=w*Wn){
				complex x=F[j+k],y=w*F[j+mid+k];
				F[j+k]=x+y,F[j+mid+k]=x-y;
			}
		}	
	}
	if(type==-1){ for(int i=0;i<limit;i++) F[i].x/=limit;}
}

int main(){
	t=rd(); int num=0;
	while(t--){
		++num;
		memset(row,0,sizeof row); memset(line,0,sizeof line); memset(qwq,0,sizeof qwq);
		n=rd()-1; m=rd()-1; cnt=rd();
		for(int i=1;i<=cnt;i++){
			int x=rd()-1,y=rd()-1; y=m-y;
			row[x]=1,line[y]=1,qwq[x+y]=1;
		}
		limit=1,l=0; while(limit<=n+m) limit<<=1,++l;
		for(int i=0;i<limit;i++) a[i].x=a[i].y=b[i].x=b[i].y=0;
		for(int i=0;i<limit;i++) rev[i]=(rev[i>>1]>>1)|((i&1)<<(l-1));
		for(int i=0;i<=n;i++)
			if(!row[i]) a[i].x=1;
		for(int i=0;i<=m;i++)
			if(!line[i]) b[i].x=1;
		FFT(a,1); FFT(b,1);
		for(int i=0;i<limit;i++) a[i]=a[i]*b[i];
		FFT(a,-1);  LL ans=0;
		for(int i=0;i<=n+m;i++){
			if(qwq[i]) continue;
			ans+=(LL)(a[i].x+0.5);
		}
		printf("Case %d: %lld\n",num,ans);
	}
	return 0;
}