航海舰队

最新推荐文章于 2018-12-26 22:14:00 发布

WerKeyTom_FTD

最新推荐文章于 2018-12-26 22:14:00 发布

阅读量1k

点赞数

分类专栏：快速傅里叶变换

本文链接：https://blog.csdn.net/WerKeyTom_FTD/article/details/70226140

版权

快速傅里叶变换专栏收录该内容

20 篇文章 0 订阅

订阅专栏

题目描述

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

空降及移动

不考虑移动，只考虑空降。
那么是个匹配问题。
用类似bzoj两个串那题的思路变成卷积形式。
可以求出那些左上角可以作为空降左上角。
这个是二维的需要展成一维。
从原位置开始bfs，一个位置如果是可移动左上角，它四个方向上的空降左上角也是可移动左上角。然后解决了移动问题。

求答案

知道了哪些可以作为可移动左上角，只要再上FFT，可以知道哪些位置可以被覆盖。

#include<cstdio>
#include<algorithm>
#include<cmath>
#define fo(i,a,b) for(i=a;i<=b;i++)
using namespace std;
typedef double db;
const db pi=acos(-1);
const int maxn=700+10,maxlen=2000000+10;
struct node{
    db x,y;
    node friend operator +(node a,node b){
        node c;
        c.x=a.x+b.x;c.y=a.y+b.y;
        return c;
    }
    node friend operator -(node a,node b){
        node c;
        c.x=a.x-b.x;c.y=a.y-b.y;
        return c;
    }
    node friend operator *(node a,node b){
        node c;
        c.x=a.x*b.x-a.y*b.y;c.y=a.x*b.y+a.y*b.x;
        return c;
    }
};
int b[maxn][maxn],a[maxn][maxn],c[maxn][maxn],d[maxn][maxn];
int rev[maxlen];
node A[maxlen],B[maxlen],C[maxlen],tt[maxlen],w[maxlen];
int dl[maxn*maxn][2],fx[5][2];
bool bz[maxn][maxn];
int i,j,k,l,t,n,m,nn,mm,ans,mix,mxx,miy,mxy,cnt,len,head,tail;
db ce;
char ch;
bool czy;
char get(){
    char ch=getchar();
    while (ch!='.'&&ch!='#'&&ch!='o') ch=getchar();
    return ch;
}
void prepare(){
    fo(i,0,len-1){
        int p=0;
        for (int j=0,tp=i;j<ce;j++,tp/=2) p=(p<<1)+(tp%2);
        rev[i]=p;
    }
    w[0].x=1;w[0].y=0;
    w[1].x=cos(2*pi/len);w[1].y=sin(2*pi/len);
    fo(i,2,len) w[i]=w[i-1]*w[1];
}
void DFT(node *a,int sig){
    int i;
    fo(i,0,len-1) tt[rev[i]]=a[i];
    for (int m=2;m<=len;m*=2){
        int half=m/2,bei=len/m;
        fo(i,0,half-1){
            node wi=sig>0?w[i*bei]:w[len-i*bei];
            /*node wi;
            wi.x=cos(i*pi*sig/half);wi.y=sin(i*pi*sig/half);*/
            for (int j=i;j<len;j+=m){
                node u=tt[j],v=tt[j+half]*wi;
                tt[j]=u+v;
                tt[j+half]=u-v;
            }
        }
    }
    if (sig==-1)
        fo(i,0,len-1) tt[i].x/=len;
    fo(i,0,len-1) a[i]=tt[i];
}
void FFT(node *a,node *b){
    int i;
    DFT(a,1);DFT(b,1);
    fo(i,0,len-1) a[i]=a[i]*b[i];
    DFT(a,-1);
}
int main(){
    freopen("sailing.in","r",stdin);freopen("sailing.out","w",stdout);
    fx[0][0]=1;fx[0][1]=0;
    fx[1][0]=-1;fx[1][1]=0;
    fx[2][0]=0;fx[2][1]=1;
    fx[3][0]=0;fx[3][1]=-1;
    scanf("%d%d",&n,&m);
    mix=miy=1000000;mxx=mxy=0;
    fo(i,1,n)
        fo(j,1,m){
            ch=get();
            if (ch=='o'){
                mix=min(mix,i);
                miy=min(miy,j);
                mxx=max(mxx,i);
                mxy=max(mxy,j);
                b[i][j]=-1;
                cnt++;
            }
            else if (ch!='#') b[i][j]=1;
        }
    nn=mxx-mix+1;mm=mxy-miy+1;
    fo(i,mix,mxx)
        fo(j,miy,mxy)
            a[i-mix+1][j-miy+1]=b[i][j];
    fo(i,1,nn)
        fo(j,1,mm)
            if (a[i][j]==-1) a[i][j]=1;else a[i][j]=0;
    fo(i,1,n)
        fo(j,1,m)
            if (b[i][j]==-1) b[i][j]=1;
    fo(i,1,nn)
        fo(j,1,mm){
            k=n-i+1;t=m-j+1;
            A[(k-1)*m+t-1].x=a[i][j];
        }
    fo(i,1,n)
        fo(j,1,m)
            B[(i-1)*m+j-1].x=b[i][j];
    len=1;
    while (len<=(n*m-1)*2) len*=2;
    ce=log(len)/log(2);
    prepare();
    FFT(A,B);
    fo(i,1,n-nn+1)
        fo(j,1,m-mm+1){
            k=i+n;t=j+m;
            if (int(A[(k-2)*m+t-2].x+0.5)==cnt) c[i][j]=1;
        }
    tail=1;
    dl[1][0]=mix;dl[1][1]=miy;
    c[mix][miy]=-1;
    while (head<tail){
        head++;
        j=dl[head][0];k=dl[head][1];
        fo(i,0,3){
            l=j+fx[i][0];t=k+fx[i][1];
            if (l<1||l>n||t<1||t>m) continue;
            if (c[l][t]==1){
                c[l][t]=-1;
                dl[++tail][0]=l;
                dl[tail][1]=t;
            }
        }
    }
    fo(i,1,n)
        fo(j,1,m)
            if (c[i][j]==-1) c[i][j]=1;else c[i][j]=0;
    fo(i,0,len-1) A[i].x=A[i].y=B[i].x=B[i].y=0;
    fo(i,1,n)
        fo(j,1,m)
            A[(i-1)*m+j-1].x=c[i][j];
    fo(i,1,nn)
        fo(j,1,mm)
            B[(i-1)*m+j-1].x=a[i][j];
    FFT(A,B);
    fo(i,1,n)
        fo(j,1,m)
            if (b[i][j]==1&&int(A[(i-1)*m+j-1].x+0.5)>0) ans++;
    printf("%d\n",ans);
}