矩阵置0

package algorithms;

/**
 * 给定一个 m x n 的矩阵，如果一个元素为 0，则将其所在行和列的所有元素都设为 0。请使用原地算法。 示例 1: 输入: [ [1,1,1],
 * [1,0,1], [1,1,1] ] 输出: [ [1,0,1], [0,0,0], [1,0,1] ] 示例 2: 输入: [ [0,1,2,0],
 * [3,4,5,2], [1,3,1,5] ] 输出: [ [0,0,0,0], [0,4,5,0], [0,3,1,0] ]
 * 
 */

//思路 第一次遍历该二维数组  
//用第0行存储每一列是否为0   用第0列存储每一行是否为0
//第二次遍历该数组  若第0列有行为0  则将该行全部置位0
//若第0行有列为0  则将该列全部置位0  
public class Exp2_1 {
    public void setZeroes(int[][] matrix) {
        boolean isRowZero = false;
        boolean isColumnZero = false;
        // 第一次遍历
        for (int i = 0; i < matrix.length; i++)
            for (int j = 0; j < matrix[0].length; j++) {
                if (matrix[i][j] == 0) {
                    // 第0行存储每一列是否为0
                    matrix[0][j] = 0;
                    // 用第0列存储每一行是否为0
                    matrix[i][0] = 0;
                    if (i == 0)
                        isRowZero = true;
                    if (j == 0)
                        isColumnZero = true;
                    break;
                }
            }
        // 第二次遍历
        for (int i = 1; i < matrix.length; i++)
            for (int j = 1; j < matrix[0].length; j++) {
                if (matrix[i][0] == 0 || matrix[0][j] == 0) {
                    matrix[i][j] = 0;
                }
            }

        // 对第0行和第0列单独处理
        if(isRowZero)
            for(int j=0;j<matrix[0].length;j++)
                matrix[0][j]=0;
        if(isColumnZero)
            for(int i=0;i<matrix.length;i++)
                matrix[i][0]=0;
    }

}