LeetCode 73. 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.

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;
    }
}