SetMatrix_0问题

数组相关元素设为0问题

原问题中,提到了让我们尽量的减少 space 的使用。这个是我们算法优化的方向。

分析

问题需要将将2维数组中的0元素的同行同列都设为0。如果不涉及 extra space 的话,可以直接申请另外一个同等大小的 二维数组来做,非常直观,但是这里由于有这个 extra space 要求,所以需要换一种方法。

思路

具体的思路呢,有点像之前的 Unique_Path 问题。就是说把首行首列和其他元素分开来处理,因为这里我们需要 把首行首列元素当做标志位,用来标志该行/列是否需要 set zero。
所以做法就分三步:

  • 先处理首行首列,看首行首列是否含0,这里用两个 extra flag 来指定
  • 处理其他元素,如果为0,将相应的首行首列元素置为0,
  • 现在首行首列变成了标志位,遍历剩下的元素,通过标志位将其他元素置0。同时根据 extra flag 把首行首列置0

代码

package day_21;

import java.util.Set;

public class SetZeros {
    public void setZeros(int[][] matrix){
        if (matrix == null || matrix.length == 0 || matrix[0].length == 0) return;
        int row = matrix.length, col = matrix[0].length;
        int flag_row = 0, flag_col=0;
        for(int i=0;i

你可能感兴趣的:(SetMatrix_0问题)