LeetCode--73--矩陣置零

題目描述:
給定一個 m x n 的矩陣,如果一個元素爲 0,則將其所在行和列的所有元素都設爲 0。請使用原地算法。
輸入:
[
[1,1,1],
[1,0,1],
[1,1,1]
]
[
[0,1,2,0],
[3,4,5,2],
[1,3,1,5]
]
輸出:
[
[1,0,1],
[0,0,0],
[1,0,1]
]
[
[0,0,0,0],
[0,4,5,0],
[0,3,1,0]
]
題意:
題目描述
題解
開一個數組記錄一下,哪些行和列需要修改,然後直接改就好了
代碼:

class Solution {
    public void setZeroes(int[][] matrix) {
        int x = matrix.length;
        int y = matrix[0].length;
        int [][] c = new int [x][y];
        for(int i = 0; i < x; i ++){
            for(int j = 0; j < y; j ++){
                if(matrix[i][j] == 0){
                    ChangeSet(i,j,c);
                }
            }
        }
        for(int i = 0; i < x; i ++){
            for(int j = 0; j < y; j ++){
                if(c[i][j] == 1){
                   matrix[i][j] = 0;
                }
            }
        }
    }

    public void ChangeSet(int col,int row,int [][] c){
        for(int i = 0; i < c.length; i ++){
            c[i][row] = 1;
        }
        for(int i = 0; i < c[0].length; i ++){
            c[col][i] = 1;
        }
    }
}
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章