JAVA程序设计：01 矩阵（LeetCode：542）

给定一个由 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。
矩阵中的元素只在四个方向上相邻: 上、下、左、右。

思路：动态规划。对于每个 1，到 0 的最短路径可能从任意方向。所以我们需要检查所有 4 个方向。在从上到下的迭代中，我们需要检查左边和上方的最短路径；我们还需要另一个从下往上的循环，检查右边和下方的方向。

class Solution {
    public int[][] updateMatrix(int[][] matrix) {
       
    	
        int m=matrix.length;
        int n=matrix[0].length;
        
        for(int i=0;i<m;i++)
        	for(int j=0;j<n;j++)
        		if(matrix[i][j]==1) matrix[i][j]=10001;
        for(int i=1;i<n;i++) matrix[0][i]=Math.min(matrix[0][i], matrix[0][i-1]+1);
        for(int i=1;i<m;i++) matrix[i][0]=Math.min(matrix[i][0], matrix[i-1][0]+1);
        for(int i=1;i<m;i++)
        	for(int j=1;j<n;j++)
        		matrix[i][j]=Math.min(matrix[i][j], Math.min(matrix[i-1][j], matrix[i][j-1])+1);
        for(int i=n-2;i>=0;i--) matrix[m-1][i]=Math.min(matrix[m-1][i], matrix[m-1][i+1]+1);
        for(int i=m-2;i>=0;i--) matrix[i][n-1]=Math.min(matrix[i][n-1], matrix[i+1][n-1]+1);
        for(int i=m-2;i>=0;i--)
        	for(int j=n-2;j>=0;j--)
        		matrix[i][j]=Math.min(matrix[i][j], Math.min(matrix[i+1][j], matrix[i][j+1])+1);
        return matrix;
    }
}