ICode9

精准搜索请尝试: 精确搜索
首页 > 其他分享> 文章详细

542. 01 矩阵

2021-12-17 23:33:00  阅读:214  来源: 互联网

标签:queue 01 mat int 矩阵 542 step new operation


给定一个由 0 和 1 组成的矩阵 mat ,请输出一个大小相同的矩阵,其中每一个格子是 mat 中对应位置元素到最近的 0 的距离。

两个相邻元素间的距离为 1 。

来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/01-matrix
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。

多源广度优先搜索

import java.util.LinkedList;
import java.util.Queue;

class Solution {

    private int[] dx = {0, 1, 0, -1};

    private int[] dy = {1, 0, -1, 0};

    public int[][] updateMatrix(int[][] mat) {
        if (mat == null || mat.length == 0 || mat[0].length == 0) {
            return new int[0][0];
        }

        int n = mat.length, m = mat[0].length;
        int[][] ret = new int[n][m];
        Queue<Operation> queue = new LinkedList<>();
        for (int i = 0; i < n; ++i) {
            for (int j = 0; j < m; ++j) {
                if (mat[i][j] == 0) {
                    queue.offer(new Operation(i, j, 0));
                }
            }
        }

        while (!queue.isEmpty()) {
            Operation operation = queue.poll();
            ret[operation.x][operation.y] = operation.step;
            for (int i = 0; i < 4; ++i) {
                int x = dx[i] + operation.x;
                int y = dy[i] + operation.y;
                if (x >= 0 && x < n && y >= 0 && y < m && mat[x][y] == 1) {
                    queue.offer(new Operation(x, y, operation.step + 1));
                    mat[x][y] = 0;
                }
            }
        }
        return ret;
    }
}

class Operation {
    int x;
    int y;
    int step;

    public Operation(int x, int y, int step) {
        this.x = x;
        this.y = y;
        this.step = step;
    }
}

动态规划


标签:queue,01,mat,int,矩阵,542,step,new,operation
来源: https://www.cnblogs.com/tianyiya/p/15704138.html

本站声明: 1. iCode9 技术分享网(下文简称本站)提供的所有内容,仅供技术学习、探讨和分享;
2. 关于本站的所有留言、评论、转载及引用,纯属内容发起人的个人观点,与本站观点和立场无关;
3. 关于本站的所有言论和文字,纯属内容发起人的个人观点,与本站观点和立场无关;
4. 本站文章均是网友提供,不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属;如您发现该文章侵犯了您的权益,可联系我们第一时间进行删除;
5. 本站为非盈利性的个人网站,所有内容不会用来进行牟利,也不会利用任何形式的广告来间接获益,纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。

专注分享技术,共同学习,共同进步。侵权联系[81616952@qq.com]

Copyright (C)ICode9.com, All Rights Reserved.

ICode9版权所有