LeetCode———数组2维：面试题 01.08. 零矩阵

最新推荐文章于 2024-08-04 16:45:57 发布

Shae Zhang

最新推荐文章于 2024-08-04 16:45:57 发布

阅读量111

点赞数

分类专栏：二维数组 LeetCode C++ 文章标签： leetcode 算法

本文链接：https://blog.csdn.net/weixin_43184354/article/details/120211419

版权

LeetCode 同时被 3 个专栏收录

13 篇文章 1 订阅

订阅专栏

C++

12 篇文章 1 订阅

订阅专栏

二维数组

1 篇文章 0 订阅

订阅专栏

一、题目

编写一种算法，若M × N矩阵中某个元素为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]
]

二、题目理解

1、遍历查找零元素，其对应的行数和列数，令该行所有元素为零，该列所有元素为零。

auto newm =  m;   % 建立一个新的数组
for(int i=0; i<n; i++){
	for(int j=0; j<n; j++){
		if(m[i][j]==0)
		{
			newm[n-1-i][0] = i;  % 对应的行存储在newm数组的第一行第一列;
			newm[n-1-i][1] = j;  % 对应的列存储在newm数组的第一行第二列;
		}
	}
}

for(int ii=0; ii<newm.size(); ii++){
	for(int jj=0; jj<2; jj++){
		if(newm[ii][jj]!=0){
			m[ii][所有列] = 0;
			m[所有行][jj] = 0;
		}
	}
}

结果不对！！
针对第一个例子：
[
[1,1,1],
[1,0,1],
[1,1,1]
]
简单写一下遍历过程。
当i = 0, j = 0; m[0][0]=1; 不为零；
当i = 0; j = 1; m[0][1]=1; 不为零
当i = 0; j = 2; m[0][2]=1; 不为零
当i = 1, j = 0; m[1][0]=1; 不为零；
当i = 1; j = 1; m[1][1]=0; 为零;
当i = 1; j = 2; m[1][2]=1; 不为零
当i = 2, j = 0; m[2][0]=1; 不为零；
当i = 2; j = 1; m[2][1]=1; 不为零
当i = 2; j = 2; m[2][2]=1; 不为零

当i = 1; j = 1; m[1][1]=0; 为零;
令[1][0] , [1][2] 还有[2][1], [0][1]
为零。

用程序表示为 A[1][j] = 0 和A[i][1] =0;

借鉴他人思路的做法

class Solution {
public:
    void setZeroes(vector<vector<int>>& matrix) {
        int row = matrix.size();     // 数组的行数
        int col = matrix[0].size();  // 数组的列数

        if(row==0 && col==0)  return;   // 若为空矩阵，返回空值


        int flag_row = 0;  // 行标记：是否为0元素
        int flag_col = 0;  // 列标记，是否有0元素
        //先扫描第一行第一列，如果有0，则将各自的flag设置为true
        for(int i=0; i<row; i++){
            if(matrix[i][0] == 0){
                flag_col = 1;
            }
        }
        for(int i=0; i<col; i++){
            if(matrix[0][i] == 0){
                flag_row = 1;
            }
        }


        //扫描除去第一行第一列的整个数组，如果有0，则将对应的第一行和第一列的数字赋0
        for(int i=0; i<row; i++){
            for(int j=0; j<col; j++){
                if(matrix[i][j] == 0){
                    matrix[0][j]=0;
                    matrix[i][0]=0;
                }
            }
        }
        // 遍历除去第一行第一列的整个数组，如果对应的第一行和第一列的数字有一个为0，则将当前值赋0
        for(int i=0; i<row; i++){
            for(int j=0; j<col; j++){
                if(matrix[i][0] == 0 || matrix[0][j] == 0){
                    matrix[i][j]=0;
                }
            }

        }

        
        // 根据第一行第一列的flag来更新第一行第一列
        if(flag_row){// 若是改该行存在零元素，那就遍历所有的列，令其为0
            for(int i=0; i<row; i++){
                matrix[0][i] = 0;
            }
        }

        if(flag_col){// 若是改该列存在零元素，那就遍历所有的行，令其为0
            for(int j=0; j<col; j++){
                matrix[j][0]=0;
            }
        }
    }
};