bzoj 5217: [Lydsy2017省队十连测]航海舰队 二维转一维FFT

题意

Byteasar 组建了一支舰队！他们现在正在海洋上航行着。海洋可以抽象成一张n×m 的网格图，其中有些位置是“.”，表示这一格是海水，可以通过；有些位置是“#”，表示这一格是礁石，不可以通过；有些位置是“o”，表示这一格目前有一艘舰，且舰离开这一格之后，这一格将变为“.”。这些“o” 表示Byteasar 的舰队，他们每天可以往上下左右中的一个方向移动一格，但不能有任何一艘舰驶出地图。特别地，Byteasar 对阵形有所研究，所以他不希望在航行的过程中改变阵形，即任何时刻任何两艘舰的相对位置都不能发生变化。Byteasar 的舰队可以航行无限长的时间，每当一艘舰经过某个格子的时候，这个格子海底的矿藏都将被Byteasar 获得。请写一个程序，帮助Byteasar 计算他最多可以获得多少个格子海底的矿藏？
$n,m<=700$

分析

首先我们把最小的包含所有”o”的矩形拿出来，设其大小为$p*q$，现在要求出对于所有大小为$p*q$的子矩形，若把舰队对应的子矩形放在此处是否会冲突。
要求这个的话，我们可以先把大矩形转成一维，具体来说就是先把第一行拿出来，然后把第二行接在第一行后面，如此类推。设转成一维后的01串为S。
接下来我们把舰队所在的$p*q$的矩形扔到一个$n*m$且全为0的矩形的左上角，然后把这个矩形也转成一维。设转成的串为T。
设$(x,y)$在A中的位置是$p$，那么以$(x,y)$为左上角的子矩形可以放下舰队当且仅当不存在一个非负整数i使得$S[p+i]=T[i]=1$。也就是说，$\sum_{i\ge0}S[p+i]*T[i]=0$。
不难发现这是一个卷积形式，把S反过来之后FFT一下判断一下每一位是否为0就可以了。
接下来bfs一下求出所有可以走到的位置，把这些位置在S上标为1，其余位置标为0。
那么$(x,y)$可以被到达当且仅当存在一个非负整数i使得$S[p-i]=T[i]=1$。
同样用FFT搞一搞即可。

代码

#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<algorithm>
#include<cmath>
#include<queue>
#define mp(x,y) make_pair(x,y)
using namespace std;

typedef pair<int,int> pa;

const int N=705;
const double pi=acos(-1);
const int inf=1000000000;

int n,m,L,rev[N*N*3],dx[4]={0,1,0,-1},dy[4]={1,0,-1,0};
bool vis[N][N];
char ma[N][N];
queue<pa> que;
struct com
{
    double x,y;
    com operator + (const com &d) const {return (com){x+d.x,y+d.y};}
    com operator - (const com &d) const {return (com){x-d.x,y-d.y};}
    com operator * (const com &d) const {return (com){x*d.x-y*d.y,x*d.y+y*d.x};}
    com operator / (const double &d) const {return (com){x/d,y/d};}
}a[N*N*3],b[N*N*3];

void fft(com *a,int f)
{
    for (int i=0;i<L;i++) if (i<rev[i]) swap(a[i],a[rev[i]]);
    for (int i=1;i<L;i<<=1)
    {
        com wn=(com){cos(pi/i),f*sin(pi/i)};
        for (int j=0;j<L;j+=(i<<1))
        {
            com w=(com){1,0};
            for (int k=0;k<i;k++)
            {
                com u=a[j+k],v=a[j+k+i]*w;
                a[j+k]=u+v;a[j+k+i]=u-v;
                w=w*wn;
            }
        }
    }
    if (f==-1) for (int i=0;i<L;i++) a[i]=a[i]/L;
}

void bfs(int sx,int sy)
{
    que.push(mp(sx,sy));
    vis[sx][sy]=0;
    while (!que.empty())
    {
        int x=que.front().first,y=que.front().second;
        a[(x-1)*m+y-1]=(com){1,0};
        que.pop();
        for (int i=0;i<4;i++)
        {
            int p=x+dx[i],q=y+dy[i];
            if (vis[p][q]) vis[p][q]=0,que.push(mp(p,q));
        }
    }
}

int main()
{
    scanf("%d%d",&n,&m);
    for (int i=1;i<=n;i++) scanf("%s",ma[i]+1);
    int x1=inf,y1=inf,x2=0,y2=0;
    for (int i=1;i<=n;i++)
        for (int j=1;j<=m;j++)
            if (ma[i][j]=='o') x1=min(x1,i),y1=min(y1,j),x2=max(x2,i),y2=max(y2,j);
            else if (ma[i][j]=='#') a[n*m-(i-1)*m-j]=(com){1,0};
    for (int i=1;i<=n;i++)
        for (int j=1;j<=m;j++)
            if (ma[i][j]=='o') b[(i-x1)*m+j-y1]=(com){1,0};
    int lg=0;
    for (L=1;L<=n*m*2;L<<=1,lg++);
    for (int i=0;i<L;i++) rev[i]=(rev[i>>1]>>1)|((i&1)<<(lg-1));
    fft(a,1);fft(b,1);
    for (int i=0;i<L;i++) a[i]=a[i]*b[i];
    fft(a,-1);
    for (int i=1;i<=n-(x2-x1);i++)
        for (int j=1;j<=m-(y2-y1);j++)
            if ((int)(a[n*m-(i-1)*m-j].x+0.1)==0) vis[i][j]=1;
    for (int i=0;i<L;i++) a[i]=(com){0,0};
    bfs(x1,y1);
    fft(a,1);
    for (int i=0;i<L;i++) a[i]=a[i]*b[i];
    fft(a,-1);
    int ans=0;
    for (int i=0;i<n*m;i++) if ((int)(a[i].x+0.1)>0) ans++;
    printf("%d",ans);
    return 0;
}