力扣73. 矩阵置零

254 阅读2分钟

Offer 驾到,掘友接招!我正在参与2022春招打卡活动,点击查看活动详情

力扣73. 矩阵置零

一、题目描述:

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

示例 1:

img

输入:matrix = [[1,1,1],[1,0,1],[1,1,1]] 输出:[[1,0,1],[0,0,0],[1,0,1]] 示例 2:

img

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

提示:

m == matrix.length

n == matrix[0].length

1 <= m, n <= 200

-2^31 <= matrixi <= 2^31 - 1

进阶:

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

来源:力扣(LeetCode) 链接:leetcode-cn.com/problems/se… 著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。

二、思路分析:

  1. 这道题考察了什么思想?你的思路是什么?

    思路很简单,就是构建两个标记数组,遍历数组matrix一次,如果某个元素是0,就把该元素那行和那列设置为true。最后再把matrix数组true位置置为0即可。

  2. 做题的时候是不是一次通过的,遇到了什么问题,需要注意什么细节?

    哈哈哈哈,这道题目是一次通过的呢~ 只需要思路清晰,这道题目应该是简单难度。

  3. 有几种解法,哪种解法时间复杂度最低,哪种解法空间复杂度最低,最优解法是什么?其他人的题解是什么,谁的效率更好一些?用不同语言实现的话,哪个语言速度最快?

    看到另外一个大佬的解法,感觉是比较巧妙的。

    image-20220320141104149

image-20220320141520924

三、AC 代码:

 class Solution {
     public void setZeroes(int[][] matrix) {
         boolean[] rows = new boolean[matrix.length];
         boolean[] columns = new boolean[matrix[0].length];
 ​
         for(int i=0;i<matrix.length;i++){
             for(int j=0;j<matrix[0].length;j++){
                 if(matrix[i][j] == 0){
                     rows[i] = true;
                     columns[j] = true;
                 }
             }
         }
 ​
         for(int i=0;i<matrix.length;i++){
             for(int j=0;j<matrix[0].length;j++){
                 if(rows[i] == true || columns[j] == true){
                     matrix[i][j] = 0;
                 }
             }
         }
     }
 }

image-20220320140755849

image-20220320140807260

 class Solution {
     public void setZeroes(int[][] matrix) {
         int row = matrix.length;
         int col = matrix[0].length;
         boolean row0_flag = false;
         boolean col0_flag = false;
         // 第一行是否有零
         for (int j = 0; j < col; j++) {
             if (matrix[0][j] == 0) {
                 row0_flag = true;
                 break;
             }
         }
         // 第一列是否有零
         for (int i = 0; i < row; i++) {
             if (matrix[i][0] == 0) {
                 col0_flag = true;
                 break;
             }
         }
         // 把第一行第一列作为标志位
         for (int i = 1; i < row; i++) {
             for (int j = 1; j < col; j++) {
                 if (matrix[i][j] == 0) {
                     matrix[i][0] = matrix[0][j] = 0;
                 }
             }
         }
         // 置0
         for (int i = 1; i < row; i++) {
             for (int j = 1; j < col; j++) {
                 if (matrix[i][0] == 0 || matrix[0][j] == 0) {
                     matrix[i][j] = 0;
                 }
             }
         }
         if (row0_flag) {
             for (int j = 0; j < col; j++) {
                 matrix[0][j] = 0;
             }
         }
         if (col0_flag) {
             for (int i = 0; i < row; i++) {
                 matrix[i][0] = 0;
             }
         } 
     }
 }

image-20220320141228444

image-20220320141241165

Go:

 func setZeroes(matrix [][]int) {
     n, m := len(matrix), len(matrix[0])
     col0 := false
     for _, r := range matrix {
         if r[0] == 0 {
             col0 = true
         }
         for j := 1; j < m; j++ {
             if r[j] == 0 {
                 r[0] = 0
                 matrix[0][j] = 0
             }
         }
     }
     for i := n - 1; i >= 0; i-- {
         for j := 1; j < m; j++ {
             if matrix[i][0] == 0 || matrix[0][j] == 0 {
                 matrix[i][j] = 0
             }
         }
         if col0 {
             matrix[i][0] = 0
         }
     }
 }

四、总结:

这道题目如果用第一种方法来解决,就对不起这道题目的中等难度的,所以建议掌握第一种解法的同时掌握第二三种方法。