BZOJ 5217: [Lydsy2017省队十连测]航海舰队

Description

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

Input

第一行包含两个正整数n;m，分别表示地图的长和宽。n, m <= 700
接下来n 行，每行有m 个字符，每个字符只能是“.”、“#”、“o” 中的一个。
输入数据保证至少有一个“o”。

Output

输出一行一个整数，即可以被经过的格子数的最大值。

Sample Input

4 5

….#

.o#.o

.o..o

..o..

Sample Output

12

分析

这题 emm 跟koi2018很像啊
就你把二维拆成一维做一遍fft就好吧

代码

#include <bits/stdc++.h>

#define mp(x,y) std::make_pair(x,y)

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

int read()
{
    int x = 0, f = 1;
    char ch = getchar();
    while (ch < '0' || ch > '9') {if (ch == '-') f = -1; ch = getchar();}
    while (ch >= '0' && ch <= '9') {x = x * 10 + ch - '0'; ch = getchar();}
    return x * f;
}

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];

int rev[N * N * 3];
char map[N][N];

int L,n,m;

void fft(com *a,int f)
{
    for (int i = 0; i < L; i++) 
        if (i < rev[i]) 
            std::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;
}

bool vis[N][N];

int dx[4] = {0,1,0,-1}, dy[4] = {1,0,-1,0};

void bfs(int sx,int sy)
{
    std::queue<std::pair<int, int> >Q;
    Q.push(mp(sx,sy));
    vis[sx][sy] = 0;
    while (!Q.empty())
    {
        int x = Q.front().first, y = Q.front().second;
        a[(x - 1) * m + y - 1] = (com){1,0};
        Q.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, Q.push(mp(p,q));
        }
    }
}

int main()
{
    n = read(), m = read();
    for (int i = 1; i <= n; i++)
        scanf("%s",map[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 (map[i][j] == 'o')
                x1 = std::min(x1, i), y1 = std::min(y1, j), x2 = std::max(x2, i), y2 = std::max(y2, j);
            else
                if (map[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 (map[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\n",ans);
}