code第一部分数组：第十九题 矩阵元素为0，横列置零

 

 

分析

O(m + n) 空间的方法很简单，设置两个数组，记录每行和每列是否存在 0。

 

void zero(int **a, int m, int n){    bool row[m], col[n];       memset(row, false, sizeof(row));    memset(col, false, sizeof(col));       for(int i=0; i
     

> &matrix){ const size_t m = matrix.size(); const size_t n = matrix[0].size(); bool row_has_zero = false; // 第一行是否存在 0 bool col_has_zero = false; // 第一列是否存在 0 for (size_t i = 0; i < n; i++) if (matrix[0][i] == 0) { row_has_zero = true; break; } for (size_t i = 0; i < m; i++) { if (matrix[i][0] == 0) { col_has_zero = true; break; } for (size_t i = 1; i < m; i++) { for (size_t j = 1; j < n; j++) { if (matrix[i][j] == 0) { matrix[0][j] = 0; matrix[i][0] = 0; } } } for (size_t i = 1; i < m; i++) { for (size_t j = 1; j < n; j++) { if (matrix[i][0] == 0 || matrix[0][j] == 0) { matrix[i][j] = 0; } } } if (row_has_zero) { for (size_t i = 0; i < n; i++) { matrix[0][i] = 0; } } if (col_has_zero) { for (size_t i = 0; i < m; i++) { matrix[i][0] = 0; } }}