P1162 填涂颜色

题目链接

https://www.luogu.com.cn/problem/P1162

题目思路

主要是找到被1围着的0，但是刚开始找的最后一个，然后WA两个点，最后找到第一个0的位置，然后宽搜成功AC

题目代码

#include <iostream>
#include <algorithm>
#include <cstring>
#include <queue>

using namespace std;
typedef pair<int, int> PII;
const int N = 50;
int g[N][N];
bool st[N][N];
int n;

void bfs(int x, int y)
{
    queue<PII> q;
    q.push({x, y});
    g[x][y] = 2;
    st[x][y] = true;
    while(q.size())
    {
        auto t = q.front();
        q.pop();
        int dx[4] = {-1, 0, 1, 0}, dy[4] = {0, 1, 0, -1};
        for(int i = 0; i < 4; i ++ )
        {
            int a = t.first + dx[i], b = t.second + dy[i];
            if(a >= 0 && a < n && b >= 0 && b < n && !st[a][b] && g[a][b] != 1)
            {
                if(g[a][b] == 0) g[a][b] = 2;
                st[a][b] = true;
                q.push({a, b});
            }
        }
    }
}

int main()
{
    cin >> n;
    int x, y, f = 0;
    for(int i = 0; i < n; i ++ )
        for(int j = 0; j < n; j ++ )
        {
            cin >> g[i][j];
        }
        
    for(int i = 0; i < n; i ++ )
    {
        for(int j = 0; j < n; j ++ )
        {
            if(g[i][j] == 1)
            {
                x = i + 1, y = j + 1;
                f = 1;
                break;
            }
        }
        if(f) break;
    }
    // cout << x << ' ' << y << endl;
    bfs(x, y);
    
    for(int i = 0; i < n; i ++ )
    {
        for(int j = 0; j < n; j ++ )
            cout << g[i][j] << ' ';
        puts("");
    }
}