LeetCode——Set Matrix Zeroes

標籤：leetcode

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

原題連結：https://oj.leetcode.com/problems/set-matrix-zeroes/

題目：給定一個m * n 的矩陣，如果有一個元素是0，將其所在行和列設為0.

思路：先記錄下是0 的元素的位置，再去置0.

public void setZeroes(int[][] matrix) {boolean firstRowZero = false,firstColumnZero = false;for(int i=0;i<matrix.length;i++){if(matrix[i][0] ==0){firstColumnZero = true;break;}}for(int i=0;i<matrix[0].length;i++){if(matrix[0][i] ==0){firstRowZero = true;break;}}for(int i=1;i<matrix.length;i++){for(int j=1;j<matrix[0].length;j++){if(matrix[i][j] ==0){matrix[i][0]=0;matrix[0][j]=0;}}}for(int i=1;i<matrix.length;i++){for(int j=1;j<matrix[0].length;j++){if(matrix[i][0] ==0 || matrix[0][j]==0){matrix[i][j]=0;}}}if(firstRowZero){for(int i=0;i<matrix[0].length;i++)matrix[0][i]=0;}if(firstColumnZero){for(int i=0;i<matrix.length;i++)matrix[i][0]=0;}}

LeetCode——Set Matrix Zeroes