Leetcode题解中级篇之数组和字符串(2)矩阵置零

题目:https://leetcode-cn.com/explore/interview/card/top-interview-questions-medium/29/array-and-strings/76/

题目描述:

给定一个 m x n 的矩阵,如果一个元素为 0,则将其所在行和列的所有元素都设为 0。请使用原地算法

示例 1:

输入: 
[
  [1,1,1],
  [1,0,1],
  [1,1,1]
]
输出: 
[
  [1,0,1],
  [0,0,0],
  [1,0,1]
]

示例 2:

输入: 
[
  [0,1,2,0],
  [3,4,5,2],
  [1,3,1,5]
]
输出: 
[
  [0,0,0,0],
  [0,4,5,0],
  [0,3,1,0]
]

进阶:

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

思路:1.遍历搜索出等于0的元素。2.标记出它们的位置。3利用标记置零

第二步的标记方法有个技巧:利用第0行和第0列 来标记等于0元素的 行和列。也就是把 matrix[i][0] 和matrix[0][j]置为0.

问题来了,如果第0行或者第0列出现了等于0的元素。那么你在用第0行或者列 进行标记置零 时 那么可能接下来整个矩阵都会变为零。所以,第0行和第0列我们要单独处理。

代码:

class Solution {
    public void setZeroes(int[][] matrix) {
      if(matrix.length==0|| matrix[0].length==0)
              return;
        /* 记录第0行的元素是否应该被置为0 */
        boolean oneRowZero = false;
        /* 记录第0列的元素是否应该被置为0 */
        boolean oneColumnZero = false;
        for(int i = 0; i < matrix.length; ++i)
        {
            for(int j = 0; j < matrix[i].length; ++j)
            {
                if(matrix[i][j] == 0)
                {
                    /* 如果某个元素是0,就将第0行的对应列,第0列的对应行置为0 */
                    matrix[i][0] = matrix[0][j] = 0;
                    /* 如果是第0行的元素是0,那么就任务第0行应该被置为0 */
                    if(i == 0)
                        oneRowZero = true;
                    /* 同理第0列 */
                    if(j == 0)
                        oneColumnZero = true;
                }   
            }   
        }   

        /* 遍历矩阵第0列,如果第0列的某个元素是0,就将对应行的元素置0 */
        /* 这里没有考虑第0行第0列 */
        for(int i = 1; i < matrix.length; ++i)
        {
            if(matrix[i][0] == 0)
            {
                for(int j = 1; j < matrix[i].length; ++j)
                    matrix[i][j] = 0;
            }
        }

        /* 同理,遍历矩阵第0行,将对应列元素置0 */
        for(int j = 1; j < matrix[0].length; ++j)
        {
            if(matrix[0][j] == 0)
            {
                for(int i = 1; i < matrix.length; ++i)
                    matrix[i][j] = 0;
            }
        }

        /* 设置第0行 */
        if(oneRowZero)
        {
            for(int j = 0; j < matrix[0].length; ++j)
                matrix[0][j] = 0;
        }
        /* 设置第0列 */
        if(oneColumnZero)
        {
            for(int i = 0; i < matrix.length; ++i)
                matrix[i][0] = 0;
        }
    }
}

反思:这道题原本我想用暴力破解的。没想到,直接超时。没想到怎么做到利用空间去解。所以看了大神的文章懂了

超级爽文地址:https://blog.csdn.net/sinat_35261315/article/details/78686967

猜你喜欢

转载自blog.csdn.net/qq_34433210/article/details/84841471