LeetCode 73. Set Matrix Zeroes

来源:互联网 发布:diy装修房子软件 编辑:程序博客网 时间:2024/05/29 18:47

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

click to show follow up.

用额外空间去记录一下哪个位置出现了0,一个记录行,一个记录列,碰到这两个索引之一直接置0

import java.util.ArrayList;public class Solution {    public void setZeroes(int[][] matrix) {        ArrayList<Integer> row = new ArrayList<Integer>();        ArrayList<Integer> col = new ArrayList<Integer>();        for(int i=0;i<matrix.length;i++){            for(int j=0;j<matrix[0].length;j++){                if(matrix[i][j]==0){                    row.add(i);                    col.add(j);                }            }        }        for(int i=0;i<row.size();i++){            for(int j=0;j<matrix[0].length;j++){                matrix[row.get(i)][j]=0;            }        }        for(int i=0;i<col.size();i++){            for(int j=0;j<matrix.length;j++){                matrix[j][col.get(i)]=0;            }        }        return;    }}


原创粉丝点击