LeetCode 73. 矩阵置零

1. 题目

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

2. 解题

2.1 O（m+n）空间复杂度

• 用数组记录每行，每列是否有0

class Solution {
public:
    void setZeroes(vector<vector<int>>& matrix) {
        int m = matrix.size(), n = matrix[0].size(), i, j;
        int r[m] = {0}, c[n] = {0};
        for(i = 0; i < m; ++i)
        {
        	for(j = 0; j < n; ++j)
        	{
        		if(matrix[i][j] == 0)
        		{
        			r[i] = 1;
        			c[j] = 1;
        		}
        	}
        }

        for(i = 0; i < m; ++i)
        {
        	for(j = 0; j < n; ++j)
        	{
        		if(r[i] == 1 || c[j] == 1)
        			matrix[i][j] = 0;
        	}
        }
    }
};

2.2 O（1）空间复杂度

• 利用两个bool变量记录第一行，第一列是否存在0
• 遍历除第一行，第一列外的元素，记录状态在矩阵的第一行，第一列
• 按照第一行，第一列的信息，将右下角矩阵处理掉
• 最后按照bool变量处理第一行，第一列

class Solution {
public:
    void setZeroes(vector<vector<int>>& matrix) {
        bool r0 = 0, c0 = 0;//记录第1行和第一列是否有0
        //遍历除了1行1列外的元素，用第1行1列作为标记位
        int i, j, m = matrix.size(), n = matrix[0].size();
        for(i = 0; i < m; ++i)
        	if(matrix[i][0] == 0)
        	{
        		c0 = 1;
        		break;
        	}

    	for(j = 0; j < n; ++j)
        	if(matrix[0][j] == 0)
        	{
        		r0 = 1;
        		break;
        	}
    	for(i = 1; i < m; ++i)
        {
        	for(j = 1; j < n; ++j)
        	{
        		if(matrix[i][j] == 0)
        		{
        			matrix[i][0] = 0;
        			matrix[0][j] = 0;
        		}
        	}
        }
        for(i = 1; i < m; ++i)
        	if(matrix[i][0] == 0)
        	{
        		for(j = 1; j < n; ++j)
        			matrix[i][j] = 0;
        	}
    	for(j = 1; j < n; ++j)
        	if(matrix[0][j] == 0)
        	{
        		for(i = 1; i < m; ++i)
        			matrix[i][j] = 0;
        	}
    	if(c0)
    		for(i = 0; i < m; ++i)
        		matrix[i][0] = 0;
    	if(r0)
	    	for(j = 0; j < n; ++j)
	        	matrix[0][j] = 0;
    }
};