【NTT】BZOJ5217 [Lydsy2017省队十连测]航海舰队

版权声明：本文为博主原创文章，未经博主允许不得转载。 https://blog.csdn.net/qq_34454069/article/details/88015554

分析：

总的来说就两步：
1、二位匹配，询问每个位置能否放置。
2、得到移动范围，求覆盖位置
两个都可以用二位 卷积做。

#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cmath>
#define SF scanf
#define PF printf
#define MAXN 710
#define MAXM 2097252
#define MOD 998244353
using namespace std;
typedef long long ll;
const int G=3;
int n,m;
char s[MAXN][MAXN];
ll fsp(ll x,int y){
	ll res=1;
	while(y){
		if(y&1)
			res=res*x%MOD;
		x=x*x%MOD;
		y>>=1;
	}
	return res;
}
void NTT(ll A[],int N,int flag){
	for(int i=1,j=0;i<N;i++){
		for(int d=N;j^=d>>=1,~j&d;);
		if(i<j)
			swap(A[i],A[j]);	
	}
	for(int i=1;i<N;i<<=1){
		ll wn=fsp(G,(MOD-1)/(i<<1));
		if(flag)
			wn=fsp(wn,MOD-2);
		for(int j=0;j<N;j+=(i<<1)){
			ll w=1;
			for(int k=0;k<i;k++,w=w*wn%MOD){
				ll x=A[j+k],y=A[i+j+k]*w%MOD;
				A[j+k]=(x+y)%MOD;
				A[i+j+k]=(x-y+MOD)%MOD;
			}
		}
	}
	if(flag){
		ll invN=fsp(N,MOD-2);
		for(int i=0;i<N;i++)
			A[i]=A[i]*invN%MOD;	
	}
}	
ll A[MAXM],B[MAXM];
int w[6][4]={{1,0},{-1,0},{0,1},{0,-1}};
bool blo[MAXN][MAXN];
void dfs(int x,int y){
	A[x*m+y]=1;
	for(int i=0;i<4;i++){
		int xx=x+w[i][0];
		int yy=y+w[i][1];
		if(blo[xx][yy]==1&&A[xx*m+yy]==0)
			dfs(xx,yy);
	}
}
int main(){
	SF("%d%d",&n,&m);
	for(int i=0;i<n;i++)
		SF("%s",s[i]);
	int tot=0,stx=0,sty=0,edx=n,edy=m;
	for(int i=0;i<n;i++)
		for(int j=0;j<m;j++)
			if(s[i][j]=='o'){
				tot++;
				stx=max(stx,i);
				sty=max(sty,j);
				edx=min(edx,i);
				edy=min(edy,j);
			}
	for(int i=0;i<n;i++)
		for(int j=0;j<m;j++){
			if(s[i][j]!='#')
				A[i*m+j]=1;
			if(s[i][j]=='o')
				B[(stx-i)*m+sty-j]=1;
		}
	int p=1;
	while(p<=2*n*m)
		p<<=1;
	NTT(A,p,0);
	NTT(B,p,0);
	for(int i=0;i<p;i++)
		A[i]=A[i]*B[i]%MOD;
	NTT(A,p,1);
	for(int i=stx-edx;i<n;i++)
		for(int j=sty-edy;j<m;j++)
			blo[i][j]=(A[i*m+j]==tot);
//	for(int i=0;i<n;i++){
//		for(int j=0;j<m;j++)
//			PF("{%d}",A[i*m+j]);
//		PF("\n");
//	}
	memset(A,0,sizeof A);
	dfs(stx,sty);
	for(int i=0;i<n;i++)
		for(int j=0;j<m;j++)
			A[i*m+j]=min(A[i*m+j],1ll*blo[i][j]);
	int empt=0;
	for(int i=0;i<p;i++)
		empt+=A[i];
	memset(B,0,sizeof B);
	for(int i=0;i<n;i++)
		for(int j=0;j<m;j++)
			if(s[i][j]=='o')
				B[i*m+j]=1;
	NTT(A,p,0);
	NTT(B,p,0);
	for(int i=0;i<p;i++)
		A[i]=A[i]*B[i]%MOD;
	NTT(A,p,1);
//	for(int i=0;i<p;i++){
//		if(i%4==0)
//			PF("\n");
//		PF("[%d]",A[i]);
//	}
	int sum=0;
	for(int i=0;i<p;i++)
		sum+=(A[i]!=0);
	PF("%d",sum);
}