一、內容
問題描述
小明有一塊空地,他將這塊空地劃分爲 n 行 m 列的小塊,每行和每列的長度都爲 1。
小明選了其中的一些小塊空地,種上了草,其他小塊仍然保持是空地。
這些草長得很快,每個月,草都會向外長出一些,如果一個小塊種了草,則它將向自己的上、下、左、右四小塊空地擴展,這四小塊空地都將變爲有草的小塊。
請告訴小明,k 個月後空地上哪些地方有草。
輸入格式
輸入的第一行包含兩個整數 n, m。
接下來 n 行,每行包含 m 個字母,表示初始的空地狀態,字母之間沒有空格。如果爲小數點,表示爲空地,如果字母爲 g,表示種了草。
接下來包含一個整數 k。
輸出格式
輸出 n 行,每行包含 m 個字母,表示 k 個月後空地的狀態。如果爲小數點,表示爲空地,如果字母爲 g,表示長了草。
樣例輸入
4 5
.g...
.....
..g..
.....
2
樣例輸出
gggg.
gggg.
ggggg
.ggg.
評測用例規模與約定
對於 30% 的評測用例,2 <= n, m <= 20。
對於 70% 的評測用例,2 <= n, m <= 100。
對於所有評測用例,2 <= n, m <= 1000,1 <= k <= 1000。
二、思路
- bfs遍歷即可。 O(nm)
三、代碼
#include <cstdio>
#include <cstring>
#include <algorithm>
#include <queue>
using namespace std;
const int N = 1005;
int n, m, k;
char g[N][N];
int dx[4] = {1, 0, 0, -1};
int dy[4] = {0, 1, -1, 0};
struct Node {
int x, y;
Node(int x, int y): x(x), y(y){}
};
bool ok(int x, int y) {
if (x < 0 || y < 0 || x >= n || y >= m || g[x][y] == 'g') return false;
return true;
}
void bfs() {
queue<Node> q;
for (int i = 0; i < n; i++) {
for (int j = 0; j < m; j++) {
if (g[i][j] == 'g') q.push(Node(i, j));
}
}
while (k) {
k--;
int len = q.size();
for (int i = 1; i <= len; i++) {
Node t = q.front(); q.pop();
int x = t.x, y = t.y;
for (int j = 0; j < 4; j++) {
int fx = x + dx[j];
int fy = y + dy[j];
if (ok(fx, fy)) {
g[fx][fy] = 'g';
q.push(Node(fx, fy));
}
}
}
}
}
int main() {
scanf("%d%d", &n, &m);
for (int i = 0; i < n; i++) {
scanf("%s", g[i]);
}
scanf("%d", &k);
bfs();
for (int i = 0; i < n; i++) printf("%s\n", g[i]);
return 0;
}