leetcode/SetMatrixZeroes.java at master · yuanx/leetcode · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
public class Solution {
    public void setZeroes(int[][] matrix) {
        // Start typing your Java solution below
        // DO NOT write main() function
		int zero_frow = 0;
		int zero_fcol = 0;

		int rows = matrix.length;
		int cols = matrix[0].length;

		for(int i=0; i<cols; i++){
			if(matrix[0][i] == 0){
				zero_frow = 1;
				break;
			}
		}

		for(int i=0; i<rows; i++){
			if(matrix[i][0] == 0){
				zero_fcol = 1;
				break;
			}
		}

		for(int i=1; i<rows; i++)
			for(int j=1; j<cols; j++){
				if(matrix[i][j] == 0){
					matrix[i][0] =0;
					matrix[0][j] =0;
				}

			}

		for(int j=1; j<cols; j++){
			if(matrix[0][j] == 0){
				for(int i=1; i<rows; i++)
					matrix[i][j] = 0;
			}
		}

		for(int i=1; i<rows; i++){
			if(matrix[i][0] ==0){
				for(int j=1; j<cols; j++)
					matrix[i][j] = 0;
			}
		}

		if(zero_frow == 1){
			for(int i=0; i<cols; i++)
				matrix[0][i] = 0;
		}

		if(zero_fcol == 1){
			for(int i=0; i<rows; i++)
				matrix[i][0] = 0;
		}


    }
}