leetcode做题总结，题目Set Matrix Zeroes 2012/04/06

这道题在cc150里见过，如果一个矩阵某个值为零，则将整行整列都清零。我的做法是用两个boolean矩阵分别记录行和列的index然后在扫一遍清零。

public void setZeroes(int[][] matrix) {
        boolean[] mb = new boolean[matrix.length];
        boolean[] nb = new boolean[matrix[0].length];
        for(int i=0;i<matrix.length;i++){
            for(int j=0;j<matrix[0].length;j++){
                if(matrix[i][j]==0){
                    mb[i]=true;
                    nb[j]=true;
                }
            }
        }
        for(int i=0;i<matrix.length;i++){
            for(int j=0;j<matrix[0].length;j++){
                if(mb[i]==true||nb[j]==true){
                    matrix[i][j]=0;
                }
            }
        }
    }

Update 2015/08/20: 上面的做法很简洁，但是多用了两个数组，下面的做法是用两个变量替换数组，实际上是使用变量记录第一列和第一行是不是0，主要是为了判断[0][0]位置是否置0，然后使用第一行和第一列来代替上面的mb和nb记录是否为0。

public class Solution {
    /**
     * @param matrix: A list of lists of integers
     * @return: Void
     */
    public void setZeroes(int[][] matrix) {
        // write your code here
        int m = matrix.length;
        if (m == 0) return;
        int n = matrix[0].length;
        if (n == 0) return;
    
        int firstrow = 1, firstcol = 1;
        for (int i = 0; i < m; i ++) 
            if (matrix[i][0] == 0) 
                firstcol = 0;
        for (int i = 0; i < n; i ++) 
            if (matrix[0][i] == 0) 
                firstrow = 0;
        for (int i = 1; i < m; i ++) {
            for (int j = 1; j < n; j ++) {
                if (matrix[i][j] == 0) {
                    matrix[i][0] = 0;
                    matrix[0][j] = 0;
                }
            }
        }
        for (int i = 1; i < m; i ++) {
            if (matrix[i][0] == 0)
                for (int j = 1; j < n; j ++) 
                    matrix[i][j] = 0;
            if (firstcol == 0) matrix[i][0] = 0;
        }
        for (int i = 1; i < n; i ++) {
            if (matrix[0][i] == 0)
                for (int j = 1; j < m; j ++) 
                    matrix[j][i] = 0;
            if (firstrow == 0) matrix[0][i] = 0;
        }
        if(firstrow == 0 || firstcol == 0) 
            matrix[0][0] = 0;
        }
}