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);
}