数组—将一个M*N矩阵中0元素所在行和列清零

Shao Kaiyang

于 2019-05-10 21:31:42 发布

阅读量1.2k

点赞数

CC 4.0 BY-SA版权

分类专栏：数据结构与算法文章标签：矩阵清零

本文链接：https://blog.csdn.net/Kaiyang_Shao/article/details/90084286

本文探讨如何高效地将M*N矩阵中0元素所在的行和列清零。通过分析避免一次性遍历导致所有元素变为0的问题，提出先标记0元素位置，再进行处理的方法。最优方案是使用两个布尔数组记录0元素的行和列，在第二遍遍历中完成清零操作。在特定场景下，还可以使用位运算进一步减小空间占用，但通常布尔类型已能满足需求。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

问题

将一个M*N矩阵中0元素所在行和列清零。

分析

注意，不能一次遍历进行处理，如果一次遍历处理最终全部的元素都会变成0，需要首先将0元素的位置标记出来，然后第二遍遍历的时候再处理；
一种可行的方式是，可以用一个大小相同的布尔数组来记录0元素出现的位置，但是这样的空间复杂度是o(nm)，其实我们不需要记录这么多值，我们仅关注0元素所在的位置，甚至我们仅关注0元素所在的行和列，因为最终不管这些行或者列上有多少个0元素都会被置为0。
所以最优的方案是，定义两个数组用来记录哪些行或者哪些列包含了0元素，第二遍遍历的时候根据行或者列包含0元素的情况进行置0。下面的实现中采用布尔类型数组进行标记，如果想再缩小空间可以利用位来表示，注意主要是标识两种状态的情况，我们都可以使用位来标记以达到缩小空间的目的。但是在实际的工程中，为了代码的易读性，且不会对空间有这么苛刻的要求情况下我们用布尔类型足够。

实现

/**
     * 1.7问题：将一个M*N矩阵中0元素所在行和列清零
     * 思路：记录0元素所在行和列，然后清零
     * time：o(mn)
     * space:o(m+n)
     */
    public static void setZero(int[][] matrix){
   
   
        boolean[