10.25算法

发布时间 2023-10-25 11:09:50作者: 卧龙丹心

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

 

示例 1:


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


输入: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
-231 <= matrix[i][j] <= 231 - 1
 

进阶:

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

暴力求解:

class Solution {
public:
    void setZeroes(vector<vector<int>>& matrix) {
    vector<pair<int,int>> zero_index;
    for(int i=0;i<matrix.size();i++){
        for(int j=0;j<matrix[i].size();j++){
            if(matrix[i][j] == 0 ){
                zero_index.push_back(make_pair(i,j));
            }
        }
    }
    for(auto x:zero_index){
        for(int i=0;i<matrix[0].size();i++){
            matrix[x.first][i] = 0;
        }
        for(int i=0;i<matrix.size();i++){
            matrix[i][x.second] = 0;
        }
    }
    }
};
 
    bool firstColHasZero = false;
    bool firstRowHasZero = false;
    for(int i=0;i<matrix[0].size();i++){
        if(matrix[0][i] == 0){
            firstRowHasZero = true;
            break;
        }
    }

    for(int i=0;i<matrix.size();i++){
        if(matrix[i][0] == 0){
            firstColHasZero = true;
            break;
        }
    }

    for(int i=1;i<matrix.size();i++){
        for(int j=1;j<matrix[0].size();j++){
            if(matrix[i][j] == 0){
                matrix[0][j] = 0;
                matrix[i][0] = 0;
            }
        }
    }

    for(int i=1;i<matrix[0].size();i++){
        if(matrix[0][i] == 0){
            for(int j=1;j<matrix.size();j++){
                matrix[j][i] = 0;
            }
        }
    }

    for(int i=1;i<matrix.size();i++){
        if(matrix[i][0] == 0){
            for(int j=1;j<matrix[0].size();j++){
                matrix[i][j] = 0;
            }
        }
    }

    if(firstColHasZero){
        for (size_t i = 0; i < matrix.size(); i++)
        {
            matrix[i][0] = 0;
        }
    }

    if(firstRowHasZero){
        for (size_t i = 0; i < matrix[0].size(); i++)
        {
            matrix[0][i] = 0;
           
        }
    }
 
 
关键:
1.暴力算法记录0的index,去让这一行和这一列为0
2.原地算法,第一行和第一列记录有0的位置,然后处理所有行列,最后处理第一行列为0的情况