从零学算法733

733.有一幅以 m x n 的二维整数数组表示的图画 image ，其中 image[i][j] 表示该图画的像素值大小。
你也被给予三个整数 sr , sc 和 newColor 。你应该从像素 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，因为它不是在上下左右四个方向上与初始点相连的像素点。
示例 2:
输入: image = [[0,0,0],[0,0,0]], sr = 0, sc = 0, newColor = 2
输出: [[2,2,2],[2,2,2]]

• 直接 dfs，从起点往四个方向延伸，每次只要遇到和起始颜色相同的就改颜色；超出边界，或者遇到和起始颜色不相同的,或者已经和要更改的颜色相同了（防止起始颜色和目标颜色相同，要是整个图画颜色都和目标颜色相同，就无限递归了，或者你也可以没 dfs 前就判断一下），就直接 return

•   int same,C,m,n;
    int[][] IMAGE;
    public int[][] floodFill(int[][] image, int sr, int sc, int color) {
        IMAGE = image;
        // 起始颜色
        same = IMAGE[sr][sc];
        C = color;
        m = IMAGE.length;
        n = IMAGE[0].length;
        dfs(sr,sc);
        return IMAGE;
    }
    public void dfs(int i,int j){
        if(i< 0 || j<0 || i>=m || j>=n || IMAGE[i][j] != same || IMAGE[i][j] == C){
            return;
        }
        IMAGE[i][j] = C;
        dfs(i+1,j);
        dfs(i-1,j);
        dfs(i,j+1);
        dfs(i,j-1);
    }
  

• 用 bfs 也大致相同，核心部分就是判断是否要改颜色的那部分代码，特判一下起始颜色等于目标颜色的情况，然后 bfs 时只要当前坐标没有超出边界并且和当前颜色不相同就改颜色然后入队

•   int m,n;
    // 模拟上下左右的坐标
    int[] dx = new int[]{0,0,-1,1};
    int[] dy = new int[]{-1,1,0,0};
    public int[][] floodFill(int[][] image, int sr, int sc, int color) {
        m = image.length;
        n = image[0].length;
        if (image[sr][sc] == color) {
            return image;
        }
        bfs(image,sr,sc,color);
        return image;
    }
    public void bfs(int[][] image,int sr, int sc, int color){
        int startColor = image[sr][sc];
        image[sr][sc] = color;
        Queue<Integer> queue = new LinkedList<>();
        queue.add(setIdx(sr,sc));
        while(!queue.isEmpty()){
            int idx = queue.poll();
            int tempX = getIdx(idx)[0];
            int tempY = getIdx(idx)[1];
            for(int i=0;i<4;i++){
                int x = tempX+dx[i];
                int y = tempY+dy[i];
                if(x >= 0 && x<m && y>=0 && y<n && image[x][y] == startColor){
                    image[x][y] = color;
                    queue.add(setIdx(x,y));
                }
            }
        }
    }
    // 二维坐标转一维数值
    public int setIdx(int x,int y){
        return x*n + y;
    }
    // 一维数值转二维坐标
    public int[] getIdx(int idx){
        return new int[]{idx/n, idx%n};
    }