給定一個由 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
注意:

給定矩陣的元素個數不超過 10000。
給定矩陣中至少有一個元素是 0。
矩陣中的元素只在四個方向上相鄰: 上、下、左、右。

思路：可以動態規劃，因為不可能出現：

拐過來形成最優解。所以正著掃一遍算兩個方向。反著掃一遍算兩個方向即可。

dp[i][j]代表這個位置的最優解。

class Solution {public int[][] updateMatrix(int[][] matrix) {int l1 = matrix.length;int l2 = matrix[0].length;if(l1 == 0) return matrix;int[][] r = new int[l1][l2];for(int i = 0; i < l1; ++i) {for(int j = 0; j < l2; ++j) {r[i][j] = 205;}}for(int i = 0; i < l1; ++i) {for(int j = 0; j < l2; ++j) {if(matrix[i][j] == 0)r[i][j] = 0;else {if(i > 0)r[i][j] = Math.min(r[i][j],r[i-1][j] + 1);if(j > 0)r[i][j] = Math.min(r[i][j],r[i][j-1] + 1);}}}for(int i = l1-1; i >= 0; --i) {for(int j = l2-1; j >= 0; --j) {if(matrix[i][j] == 0)r[i][j] = 0;else {if(i < l1-1)r[i][j] = Math.min(r[i][j],r[i+1][j] + 1);if(j < l2-1)r[i][j] = Math.min(r[i][j],r[i][j+1] + 1);}}}return r;} }

總結

以上是生活随笔為你收集整理的leetcode542 01矩阵的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。