LeetCode Hot 100 矩阵置零

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

    示例 1:

    LeetCode Hot 100 矩阵置零_第1张图片

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

    示例 2:

    LeetCode Hot 100 矩阵置零_第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>& matrix) {
            int m = matrix.size();
            int n = matrix[0].size();
            int flag = false;
            for(int i = 0;i < m;i++)
            {
                if(!matrix[i][0])
                {
                    flag = true;
                }
    
                for(int j =  1;j=0; i--)
            {
                for(int j = 1;j < n; j++)
                {
                    if(!matrix[i][0] || !matrix[0][j])
                    {
                        matrix[i][j] = 0;
    
                    }
                }
    
                if(flag)
                {
                    matrix[i][0] = 0;
                }
            }
            
        }
    };

    你可能感兴趣的:(leetcode,矩阵,算法)