程序员面试金典-面试题 01.08. 零矩阵

题目：

编写一种算法，若M × N矩阵中某个元素为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即可。

程序：

class Solution {
    public void setZeroes(int[][] matrix) {
        int row = -1;
        boolean delr = false;
        for(int i = 0; i < matrix.length; ++i){
            for(int j = 0; j < matrix[i].length; ++j){
                if(matrix[i][j] == 0){
                    row = i;
                    break;
                }
            }
            if(row != -1)
                break;
        }
        if(row == -1)
            return;
        for(int i = 0; i < matrix.length; ++i){
            for(int j = 0; j < matrix[i].length; ++j){
                if(matrix[i][j] == 0){
                    delr = true;
                    matrix[row][j] = 0;
                }
            }
            if(i != row && delr){
                for(int k = 0; k < matrix[i].length; ++k){
                    matrix[i][k] = 0;
                }
            }
            delr = false;
        }
        for(int i = 0; i < matrix[row].length; ++i){
            if(matrix[row][i] == 0){
                for(int j = 0; j < matrix.length; ++j){
                    matrix[j][i] = 0;
                }
            }else{
                matrix[row][i] = 0;
            }
        }
    }
}