LeetCode 73. Set Matrix Zeroes

Given a m x n matrix, if an element is 0, set its entire row and column to 0. Do it in place.

题目大意就是将矩阵中等于0的点所在的行和列中的元素全部置为0;

并且要求不得声明额外的数组空间。

既然不得在额外的数组空间中记录, 那么就在本身的数组中记录。

  • 用两个变量xy分别标记第一行和第一列是否出现过0
  • 然后就可以用矩阵的第一行和第一列来记录矩阵中出现0的行和列
  • 最后通过将记录的行和列全部置为0
  • 然后由xy的值判断第一行和第一列是否置为0
class Solution {
public:
    void setZeroes(vector<vector<int>>& matrix) {
        
        int row = matrix.size(), cal = matrix[0].size();
        bool x = false, y = false;
        for(int i=0; i<row; ++ i)
        {
            if(matrix[i][0] == 0)
            {
                x = true;
                break;
            }
        }
        for(int j=0; j<cal; ++ j)
        {
            if(matrix[0][j] == 0)
            {
                y = true;
                break;
            }
        }
        
        for(int i=1; i<row; ++ i)
        {
            for(int j=1; j<cal; ++ j)
            {
                if(matrix[i][j] == 0)
                {
                    matrix[i][0] = 0;
                    matrix[0][j] = 0;
                }
            }
        }
        
        for(int i=1; i<row; ++ i)
        {
            if(matrix[i][0] == 0)
            {
                for(int j=0; j<cal; ++ j)
                    matrix[i][j] = 0;
            }
        }
        for(int i=1; i<cal; ++ i)
        {
            if(matrix[0][i] == 0)
            {
                for(int j=0; j<row; ++ j)
                    matrix[j][i] = 0;
            }
        }
        
        if(x)
        {   
            for(int i=0; i<row; ++ i)
                matrix[i][0] = 0;
        }
        if(y)
        {
            for(int i=0; i<cal; ++ i)
                matrix[0][i] = 0;
        }
    }
};
posted @ 2017-04-13 19:36  aiterator  阅读(99)  评论(0编辑  收藏  举报