leetcode - Set Matrix Zeroes

标签：style   color   io   ar   for   sp   on   amp   line   

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

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

leetcode - Set Matrix Zeroes

标签：style   color   io   ar   for   sp   on   amp   line   

原文地址：http://blog.csdn.net/akibatakuya/article/details/40046239