矩阵置零的两种思路及JAVA实现

package arrayandstring;

import java.util.Arrays;
import java.util.HashSet;
//一个直接的解决方案是使用  O(mn) 的额外空间,但这并不是一个好的解决方案。
//一个简单的改进方案是使用 O(m + n) 的额外空间,但这仍然不是最好的解决方案。
//你能想出一个常数空间的解决方案吗?

public class SetMatrixZeroes {
    public static void main(String[] args) {
        int[][] matrix = {
                {0,1,2,0},
                {3,4,5,2},
                {1,3,1,5}
        };
        SetMatrixZeroes test = new SetMatrixZeroes();

        test.setZeroes(matrix);

        for (int[] link : matrix) {
            System.out.println(Arrays.toString(link));
        }

    }
    public void setZeroes(int[][] matrix) {
        int n = matrix.length;
        int m = matrix[0].length;
        //O(m + n) 的额外空间
        HashSet<Integer> row = new HashSet<>(n);
        HashSet<Integer> col = new HashSet<>(m);

        for(int i = 0; i < n; i ++) {
            for (int j = 0; j < m; j ++) {
                if (matrix[i][j] == 0) {
                    row.add(i);
                    col.add(j);
                }
            }
        }

        //设置行
        /*Iterator it = row.iterator();
        while (it.hasNext()) {
            Object r = it.next();
            int
        }*/
        for (int r : row) {
            for (int i = 0; i < m; i ++) {
                matrix[r][i] = 0;
            }
        }

        for (int c : col) {
            for (int i = 0; i < n; i ++) {
                matrix[i][c] = 0;
            }
        }
    }

    public void setZeroes02(int[][] matrix) {
        int R = matrix.length;
        int C = matrix[0].length;

        int MODIFIED = -1000000;//标志要置为0的地方

        for (int i = 0; i < R; i ++) {
            for (int j = 0; j < R; j ++) {
                if (matrix[i][j] == 0) {
                    for (int k = 0; k < C; k ++) {
                        if (matrix[i][k] != 0) {
                            matrix[i][k] = MODIFIED;
                        }
                    }
                    
                    for (int k = 0; k < R; k ++) {
                        if (matrix[k][j] != 0) {
                            matrix[k][j] = MODIFIED;
                        }
                    }
                }
            }
        }

        for (int i = 0; i < R; i ++) {
            for (int j = 0; j < R; j++) {
                if (matrix[i][j] == MODIFIED) {
                    matrix[i][j] = 0;
                }
            }
        }
        
    }
}

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值