Set Matrix Zeroes

来源:互联网 发布:mvc框架 java 编辑:程序博客网 时间:2024/06/16 12:20

Set Matrix Zeroes


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

Java代码:

public class Solution {    public void setZeroes(int[][] matrix) {        boolean firstColZero = false, firstRowZero = false;        for(int i = 0;i < matrix.length;i++)            if(matrix[i][0] == 0)                firstColZero = true;        for(int j = 0;j < matrix[0].length;j++)            if(matrix[0][j] == 0)                firstRowZero = true;        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] = matrix[0][j] = 0;        for(int i = 1;i < matrix.length;i++)            if(matrix[i][0] == 0)                for(int j = 0;j < matrix[0].length;j++)                    matrix[i][j] = 0;        for(int j = 1;j < matrix[0].length;j++)            if(matrix[0][j] == 0)                for(int i = 0;i < matrix.length;i++)                    matrix[i][j] = 0;        if(firstColZero)            for(int i = 0;i < matrix.length;i++)                matrix[i][0] = 0;        if(firstRowZero)            for(int j = 0;j < matrix[0].length;j++)                matrix[0][j] = 0;    }}

 

0 0
原创粉丝点击