力扣73置零矩阵(常量空间解决方法)

1、https://leetcode-cn.com/problems/set-matrix-zeroes/

2、题目

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

进阶:

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

3、示例

示例 1:

力扣73置零矩阵(常量空间解决方法)

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

这题用常数空间的话有一个很关键的点

要留出第一行第一列出来记录中间0的位置

然后另一个for还是要从第二行第二列开始,把除了第一行第一列的地方该置0置0  (不然我可能第一行第一列就被全部置0了就没法做了,这里就是靠着第一行第一列的空间来节约空间)

最后再管第一行第一列

class Solution {
public:
    void setZeroes(vector<vector<int>>& matrix) {
        int m = matrix.size();
        int n = matrix[0].size();
        bool row = false;
        bool line = false;
        for(int i = 0; i < m; i++)
        {
            if(matrix[i][0] == 0)
            {
                line = true;
            }
        }
        for(int i = 0; i < n; i++)
        {
            if(matrix[0][i] == 0)
            {
                row == true;//第一行全置为0
            }
        }
        for(int i = 1; i < m; i++)
        {
            for(int j = 1; j < n; j++)
            {
                if(matrix[i][j] == 0)
                {
                    matrix[i][0] = 0;
                    matrix[0][j] = 0;
                }
            }
        }
        for(int i = 1; i < m; i++)
        {
            for(int j = 1; j < n; j++)
            {
                if(matrix[i][0] == 0 || matrix[0][j] == 0)
                {
                    matrix[i][j] = 0;
                }
            }
        }
        if(row)
        {
            for(int i = 0; i < n; i++)
            {
                matrix[0][i] = 0;
            }
        }
        if(line)
        {
            for(int i = 0; i < m; i++)
            {
                matrix[i][0] = 0;
            }
        }
    }
};

 

 

我的思路我只想到了mn的空间复杂度,我用set保存含0的行和列:这里用了set就可以去接着用迭代器去遍历对应的行和列的set元素,我还转了个vector这样太浪费空间,

class Solution {
public:
set<int>row;
set<int>line;
    void setZeroes(vector<vector<int>>& matrix) {
        for(int i = 0; i < matrix.size(); i++)
        {
            for(int j = 0; j < matrix[0].size(); j++)
            {
                if(matrix[i][j] == 0)
                {
                    row.insert(i);
                    line.insert(j);
                }
            }
        }
        // vector<int>row1;
        // vector<int>line1;
        // for(auto c : row)
        // {
        //     row1.push_back(c);
        // }
        // for(auto c : line)
        // {
        //     line1.push_back(c);
        // }
        // sort(row1.begin(),row1.end());
        // sort(line1.begin(),line1.end());

        for(auto it = row.begin(); it != row.end(); it++)
        {
            for(int j = 0; j < matrix[0].size(); j++)
            {
                matrix[*it][j] = 0;
            }
        }
        for(auto it = line.begin(); it != line.end(); it++)
        {
            for(int i = 0; i < matrix.size(); i++)
            {
                matrix[i][*it] = 0;
            }
        }
    }
};

 

上一篇:栈的应用:每日温度


下一篇:省选模拟73 题解