[算法练习及思路-程序员面试金典(Java解法)]No50.颜色填充

题号:no50

题目名:颜色填充

原题URL:https://leetcode-cn.com/problems/color-fill-lcci/

题目描述

编写函数，实现许多图片编辑软件都支持的「颜色填充」功能。

待填充的图像用二维数组 image 表示，元素为初始颜色值。初始坐标点的横坐标为 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] 内。
• 初始坐标点 (sr,sc) 满足 0 <= sr < image.length 和 0 <= sc < image[0].length 。
• image[i][j] 和 newColor 表示的颜色值在范围 [0, 65535] 内。

思路

1.给定一个二维坐标轴,和一个中心点,向四周扩散,如果和初始点相同则改变颜色

2.根据中心点扩散思想,考虑使用bfs向外扩散

解题代码

class Solution {
    public int[][] floodFill(int[][] image, int sr, int sc, int newColor) {
        //考虑使用bfs
        //标记是否使用过该位置
        int[][] isUsed = new int[image.length][image[0].length];
        int oldColor = image[sr][sc];
        Deque<int[]> queue = new ArrayDeque<>();
        queue.addLast(new int[]{sr,sc});
        image[sr][sc] = newColor;
        isUsed[sr][sc] = 1;
        while (!queue.isEmpty()) {
            int size = queue.size();
            while (size!=0) {
                int[] ints = queue.removeFirst();
                image[ints[0]][ints[1]] = newColor;
                isUsed[ints[0]][ints[1]] = 1;
                //bfs,上下左右进行扫描
                //如果可以向上
                if(ints[0]-1>=0 && image[ints[0]-1][ints[1]]==oldColor && isUsed[ints[0]-1][ints[1]]==0) {
                    queue.addLast(new int[]{ints[0]-1,ints[1]});
                }
                //如果可以向下
                if(ints[0]+1 < image.length && image[ints[0]+1][ints[1]]==oldColor && isUsed[ints[0]+1][ints[1]]==0) {
                    queue.addLast(new int[]{ints[0]+1,ints[1]});
                }
                //如果可以向左
                if(ints[1]-1 >= 0 && image[ints[0]][ints[1]-1]==oldColor && isUsed[ints[0]][ints[1]-1]==0) {
                    queue.addLast(new int[]{ints[0],ints[1]-1});
                }
                //如果可以向右
                if(ints[1]+1 < image[0].length && image[ints[0]][ints[1]+1]==oldColor && isUsed[ints[0]][ints[1]+1]==0) {
                    queue.addLast(new int[]{ints[0],ints[1]+1});
                }
                size--;
            }
        }
        return image;
    }
}