Leetcode 73.矩阵置零（中级算法-数组）

题目描述

给定一个 m x n 的矩阵，如果一个元素为 0，则将其所在行和列的所有元素都设为 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]
]

进阶:

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

求解历程

1、第一次通过时的思路

简单暴力地用非常数空间过了，即定义两个数组分别用于保存需要置为0的行数和列数，一次遍历求得两个数组，然后分别置零。其中在遍历行时可以直接在判断出0元素后把这行后面的元素直接置0，在后面置0时只需要把0元素前面的元素置0即可，省下一点时间。这个需要O(m+n)的额外空间。

class Solution {
public:
    void setZeroes(vector<vector<int>>& matrix) {
        vector<int> n,c;    //保存行、列的零元素
        int flag;
        for(int i=0;i<matrix.size();i++)	//遍历矩阵
        {
            flag=0;
            for(int j=0;j<matrix[i].size();j++)
            {
                if(matrix[i][j]==0)
                {
                    flag=1;
                    n.push_back(i);		//存放需要置0的行列
                    c.push_back(j);
                }
                else if(flag)
                {
                    matrix[i][j]=0;
                }
            }
        }
        vector<int>::iterator it=c.begin();
        for(;it!=c.end();it++)
        {
            for(int i=0;i<matrix.size();i++)matrix[i][*it]=0;	//列置为0
        }
        it=c.begin();
        vector<int>::iterator it1=n.begin();
        for(;it1!=n.end();it1++,it++)
        {
            for(int i=*it;i>=0;i--)matrix[*it1][i]=0;	//行置为0
        }  
    }
};

2、给出的最好范例的思路

如何在原地将矩阵置为0？可以知道的是你不能直接遍历然后遇到0就把所在行和列都置为0，因为在你置为0的同时引入了新的0，这样就不知道那些0是原来矩阵的，最终结果肯定是把所有的元素都置为0了。因此得临时存放需要置0的行和列这两种数据。巧妙的是，可以利用原始矩阵的第一行和第一列存放它们。

(1)用矩阵的第一行和第一列来存放需要置0的行和列，还需要两个布尔型变量，分别遍历第一行和第一列判断是否有0元素，有则更新对应的布尔变量。
(2)然后遍历除去第一行和第一列的矩阵，如果遇到0，则将第一行第j列的元素置为0，将第i行第一列的元素置为0。
(3)接着分别遍历第一行（和第一列），如遇到0则将该0所在列（所在行）置为0。
(4)最后判断两个布尔值是否为真，真则将第一行（第一列）置为0。

class Solution {
public:
    void setZeroes(vector<vector<int>>& matrix) {
        int m = matrix.size(), n = matrix[0].size(), up = max(n, m);
        bool firstRow = false, firstColumn = false;
        for (int i = 0; i < m; i++) {
            if (matrix[i][0] == 0) {
                firstColumn = true;		//第一行有0元素则置为真
                break;
            }
        }
        for (int i = 0; i < n; i++) {
            if (matrix[0][i] == 0) {
                firstRow = true;		//第一列有0元素则置为真
                break;
            }
        }
        for (int i = 1; i < m; i++) {
            for (int j = 1; j < n; j++) {	//遍历除去第一行和第一列的矩阵
                if (matrix[i][j] == 0)	//如果遇到0，则将第一行第j列的元素置为0，将第i行第一列的元素置为0
                    matrix[0][j] = matrix[i][0] = 0;
            }
        }
        for (int i = 1; i < m; i++) {		//遍历第一行，如遇到0则将该0所在列置为0。
            if (matrix[i][0] == 0) {
                for (int j = 0; j < n; j++)
                    matrix[i][j] = 0;
            }
        }
        for (int i = 1; i < n; i++) {		//遍历第一列，如遇到0则将该0所在行置为0。
            if (matrix[0][i] == 0) {
                for (int j = 0; j < m; j++)
                    matrix[j][i] = 0;
            }
        }
        if (firstColumn)	//判断两个布尔值是否为真，真则将第一行（第一列）置为0。
            for (int i = 0; i < m; i++)
                matrix[i][0] = 0;
        if (firstRow)
            for (int i = 0; i < n; i++)
                matrix[0][i] = 0;
        return ;
    }
};