Rayman的绝顶之路——Leetcode每日一题打卡1

Leetcode542. 01 矩阵 题目:

给定一个由 0 和 1 组成的矩阵,找出每个元素到最近的 0 的距离。
两个相邻元素间的距离为 1 。

示例 1:
输入:

0 0 0
0 1 0
0 0 0

输出:

0 0 0
0 1 0
0 0 0

示例 2:
输入:

0 0 0
0 1 0
1 1 1

输出:

0 0 0
0 1 0
1 2 1

注意:

  1. 给定矩阵的元素个数不超过 10000。
  2. 给定矩阵中至少有一个元素是 0。
  3. 矩阵中的元素只在四个方向上相邻: 上、下、左、右。

想法思路:

  1. 首先得到一个0,1二维数组,创建一个与该二维数组一样大小的二维数组并初始化,初始化时对该二维数组原本为0的地方设为0,原本有数字的地方设为一个最大值(上限+1)
  2. 然后进行第一次遍历,从左到右、从上到下重新赋值;
  3. 然后进行第二次遍历,从右到左、从下到上重新赋值。

具体代码实现如下所示:

import java.util.Arrays;

public class Leetcode542_01 {
    static class Solution {
        int[][] updateMatrix(int[][] matrix) {
            if (matrix == null || matrix.length == 0) return null;
            int m = matrix.length, n = matrix[0].length;
            int[][] res = new int[m][n];
            //初始化
            for (int i = 0; i < m; i++) {
                Arrays.fill(res[i], 10001);
            }
            for (int i = 0; i < m; i++) {
                for (int j = 0; j < n; j++) {
                    if (matrix[i][j] == 0) res[i][j] = 0;
                }
            }
            //动态规划
            //从左到右、从上到下重新赋值
            for (int i = 0; i < m; i++) {
                for (int j = 0; j < n; j++) {
                    if (i - 1 >= 0) {
                        res[i][j] = Math.min(res[i][j], res[i - 1][j] + 1);
                    }
                    if (j - 1 >= 0) {
                        res[i][j] = Math.min(res[i][j], res[i][j - 1] + 1);
                    }
                }
            }
            //从右到左、从下到上重新赋值
            for (int i = m - 1; i >= 0; i--) {
                for (int j = n - 1; j >= 0; j--) {
                    if (i + 1 < m) {
                        res[i][j] = Math.min(res[i][j], res[i + 1][j] + 1);
                    }
                    if (j + 1 < n) {
                        res[i][j] = Math.min(res[i][j], res[i][j + 1] + 1);
                    }
                }
            }
            return res;
        }
    }

    public static void main(String[] args) {
        Solution solution = new Solution();
        int[][] a = {
                {0, 0, 0},
                {0, 1, 0},
                {1, 1, 1}};
        int[][] b = solution.updateMatrix(a);
        for (int i = 0; i < b.length; i++) {
            for (int j = 0; j < b[0].length; j++) {
                System.out.print(b[i][j] + " ");
            }
            System.out.println();
        }
    }
}

2020.4.15打卡

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值