[LeetCode]Set Matrix Zeroes

Question
Given a m x n matrix, if an element is 0, set its entire row and column to 0. Do it in place.

本题难度Medium。

集合法

【复杂度】
时间 O(MN) 空间 O(1)

【思路】
如果​​​matrix[i][j]==0​​，那么就对其所在的行列上的元素置0。不过要注意这里仅对该元素的上面和左边的元素置0，对于下面的和右边的不进行置0而是将其行列分别存入Set中。原因在于上面和左边的元素置0不会对后面的判断有影响，而下面的和右边的元素原本可能不是0，这下你置0了就会产生错判（新的置0元素并不会导致对其所在的行列上的元素置0）。例如：

input：
0 0 0 5
4 3 1 4 
0 1 1 4
1 2 1 3
对于matrix[0][0]==0,那么如果将该行列元素全部置0：
0 0 0 0
0 3 1 4 
0 1 1 4
0 2 1 3
然后matrix[1][0]==0，matrix[2][0]==0，matrix[3][0]==0
导致：
0 0 0 0
0 0 0 0
0 0 0 0
0 0 0 0

【注意】
本题要求：Do it in place.

【代码】

public class Solution {
    public void setZeroes(int[][] matrix) {
        //require
        int m=matrix.length;
        if(m<1)
            return;
        int n=matrix[0].length;
        Set<Integer> rowSet=new HashSet<Integer>(),colSet=new HashSet<Integer>();
        //invariant
        for(int i=0;i<m;i++)
            for(int j=0;j<n;j++){
                if(matrix[i][j]==0){
                    for(int x=0;x<i;x++)
                        matrix[x][j]=0;
                    for(int x=0;x<j;x++)
                        matrix[i][x]=0;
                    rowSet.add(i);colSet.add(j);
                }else{
                    if(rowSet.contains(i)||colSet.contains(j))
                        matrix[i][j]=0;
                }
            }
    }
}