BFS算法 蓝桥杯长草问题

题目描述

小明有一块空地，他将这块空地划分为 nn 行 mm 列的小块，每行和每列的长度都为 1。

小明选了其中的一些小块空地，种上了草，其他小块仍然保持是空地。

这些草长得很快，每个月，草都会向外长出一些，如果一个小块种了草，则它将向自己的上、下、左、右四小块空地扩展，

这四小块空地都将变为有草的小块。请告诉小明，kk 个月后空地上哪些地方有草。

解题思路

见 BFS算法入门（走迷宫问题）

代码

#include <bits/stdc++.h>
using namespace std;
char a[10000][10000];
struct node {
    int x;
    int y;
};
queue<node> q;
int dx[] = {0, 1, 0, -1}; //左右上下
int dy[] = {1, 0, -1, 0}; //左右上下
int main() {
    int n, m;
    cin >> n >> m;
    for (int i = 0; i < n; i++) {
        for (int j = 0; j < m; j++) {
            cin >> a[i][j];
            if (a[i][j] == 'g') {
                node newNode;
                newNode.x = i;
                newNode.y = j;
                q.push(newNode);
            }
        }
    }
    int k;
    cin >> k;
    while (k--) {
        int len = q.size();
        while (len--) {
            int x = q.front().x;
            int y = q.front().y;
            for (int i = 0; i < 4; i++) {
                int tx = x + dx[i]; //周边节点的X轴
                int ty = y + dy[i]; //周边节点的y轴
                if (a[tx][ty] == '.') {
                    a[tx][ty] = 'g';
                    node newNode;
                    newNode.x = tx;
                    newNode.y = ty;
                    q.push(newNode);
                }
            }
            q.pop();
        }
    }
    for (int i = 0; i < n; i++)
        for (int j = 0; j < m; j++)
            cout << a[i][j];
        cout << endl;
    return 0;
}