涂色-BFS

Posted 2020-12-08 52dxer

涂色

有一幅以二维整数数组表示的图画，每一个整数表示该图画的像素值大小，数值在 0 到 65535 之间。给你一个坐标 (sr, sc) 表示图像渲染开始的像素值（行 ，列）和一个新的颜色值 newColor，让你重新上色这幅图像。为了完成上色工作，从初始坐标开始，记录初始坐标的上下左右四个方向上像素值与初始坐标相同的相连像素点，接着再记录这四个方向上符合条件的像素点与他们对应四个方向上像素值与初始坐标相同的相连像素点，……，重复该过程。将所有有记录的像素点的颜色值改为新的颜色值。最后返回经过上色渲染后的图像。

示例 1:

输入:

image = [
   [1,1,1],
   [1,1,0],
   [1,0,1]]
sr = 1, sc = 1, newColor = 2

输出:

[[2,2,2],
 [2,2,0],
 [2,0,1]]

解析:

在图像的正中间，(坐标(sr,sc)=(1,1)),在路径上所有符合条件的像素点的颜色都被更改成2。注意，右下角的像素没有更改为2，因为它不是在上下左右四个方向上与初始点相连的像素点。

注意:

image 和 image[0] 的长度在范围 [1, 50] 内。给出的初始点将满足 0 <= sr < image.length 和 0 <= sc < image[0].length。
image[i][j] 和 newColor 表示的颜色值在范围 [0, 65535]内。

思路：对（sr,sc）点进行bfs即可

#include <iostream>
#include <math.h>
#include <queue>
#include <string.h>
using namespace std;

// 方向数组 上右下左
int dx[4] = {-1, 0, 1, 0};
int dy[4] = {0, 1, 0, -1};
// 地图
int nums[55][55];
// 是否访问过   0代表没访问过，1代表访问过
int vis[55][55];
// 地图长度
int n;
int sr, sc, newColor;

bool check(int x, int y)
{
    return x >= 0 && y >= 0 && x < n && y < n && vis[x][y] == 0 && nums[x][y] == nums[sr][sc];
}

void bfs(int x, int y)
{
    // res[x][y] = 0;
    queue<pair<int, int>> q;
    memset(vis, 0, sizeof(vis));
    vis[x][y] = 1; //设置(x,y)访问过
    q.push({x, y});

    while (!q.empty())
    {
        // len++;
        pair<int, int> t = q.front();
        q.pop();
        for (int i = 0; i < 4; i++)
        {
            int newx = t.first + dx[i];
            int newy = t.second + dy[i];
            if (check(newx, newy))
            {
                // 将新点标记访问过
                vis[newx][newy] = 1;
                // 更新该点的像素值
                nums[newx][newy] = newColor;
                q.push({newx, newy});
            }
        }
    }
}

int main()
{
    cin >> n;
    for (int i = 0; i < n; i++)
        for (int j = 0; j < n; j++)
            cin >> nums[i][j];

    cin >> sr >> sc >> newColor;
    bfs(sr, sc);

    nums[sr][sc] = newColor;
    cout << endl;
    for (int i = 0; i < n; i++)
    {
        for (int j = 0; j < n; j++)
            cout << nums[i][j] << " ";
        cout << endl;
    }

    return 0;
}