leetcode:Set Matrix Zeors 菜鸟解法

来源:互联网 发布:大数据主要应用领域 编辑:程序博客网 时间:2024/06/07 02:43

这题本来想的挺好,觉得只要两个数组,一个记录哪行出现过0,一个记录哪列出现个0就好了,结果发现原来不能额外空间...


所以用第一行和第一列代替了,先查看第一列与第一行是否有0元素,之后可按之前的行列数组法对剩余数组进行处理。

代码如下:

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

0 0
原创粉丝点击