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

分析：

总的来说就两步：
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];
int fsp(int x,int y){
	int res=1;
	while(y){
		if(y&1)
			res=1ll*res*x%MOD;
		x=1ll*x*x%MOD;
		y>>=1;
	}
	return res;
}
int W1[30],W2[30];
void NTT(int 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,id=0;i<N;i<<=1,id++){
		int wn=W1[id];
		if(flag)
			wn=W2[id];
		for(int j=0;j<N;j+=(i<<1)){
			int w=1;
			for(int k=0;k<i;k++,w=1ll*w*wn%MOD){
				ll x=A[j+k],y=1ll*A[i+j+k]*w%MOD;
				A[j+k]=(x+y)%MOD;
				A[i+j+k]=(x-y+MOD)%MOD;
			}
		}
	}
	if(flag){
		int invN=fsp(N,MOD-2);
		for(int i=0;i<N;i++)
			A[i]=1ll*A[i]*invN%MOD;	
	}
}	
int A[MAXM],B[MAXM],A1[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(){
//	freopen("sailing.in","r",stdin);
//	freopen("sailing.out","w",stdout);
	for(int i=0;i<20;i++){
		W1[i]=fsp(G,(MOD-1)>>(i+1));
		W2[i]=fsp(W1[i],MOD-2);
	}
	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<=n*m)
		p<<=1;
	NTT(A,p,0);
	NTT(B,p,0);
	for(int i=0;i<p;i++)
		A[i]=1ll*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++)
			A1[(n-i-1)*m+m-j-1]=min(A[i*m+j],int(blo[i][j]));
	int empt=0;
	for(int i=0;i<p;i++)
		empt+=A1[i];
	NTT(A1,p,0);
	for(int i=0;i<p;i++)
		A1[i]=1ll*A1[i]*B[i]%MOD;
	NTT(A1,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+=(A1[i]!=0);
	PF("%d",sum);
}