《leetCode》：Set Matrix Zeroes

最新推荐文章于 2025-09-11 19:19:55 发布

HelloWorld_EE

最新推荐文章于 2025-09-11 19:19:55 发布

阅读量480

点赞数 1

CC 4.0 BY-SA版权

分类专栏： leetCode 文章标签： matrix leetcode zeroes

本文链接：https://blog.youkuaiyun.com/u010412719/article/details/50099513

leetCode 专栏收录该内容

205 篇文章

订阅专栏

本文提供了一种在不使用额外空间的情况下，实现矩阵中元素为零时将其所在行和列置零的方法。通过使用布尔数组记录需要置零的行和列，实现了O(1)空间复杂度的解决方案。

题目

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


Follow up:
Did you use extra space?
A straight forward solution using O(mn) space is probably a bad idea.
A simple improvement uses O(m + n) space, but still not the best solution.
Could you devise a constant space solution?

题目大意：在一个m*n的矩阵matrix中，如果matrix[i][j]=0,则将第i行和第j列的全部设置为零;

思路一

暴力搜索法：A simple improvement uses O(m + n) space, but still not the best solution.


#include<stdio.h>
#include<stdlib.h>
#include<string.h>

void setZeroes(int** matrix, int matrixRowSize, int matrixColSize) {
    if(matrix==NULL||matrixRowSize<1||matrixColSize<1){
        return;
    }
    //用来记录哪些行那些列要设为零。
    bool *row=(bool *)malloc(matrixRowSize*sizeof(bool));
    bool *col=(bool *)malloc(matrixColSize*sizeof(bool));
    if(row==NULL||col==NULL){
        exit(EXIT_FAILURE);
    }
    memset(row,true,matrixRowSize*sizeof(bool));
    memset(col,true,matrixColSize*sizeof(bool));
    for(int i=0;i<matrixRowSize;i++){
        for(int j=0;j<matrixColSize;j++){
                if(matrix[i][j]==0){
                    row[i]=false;
                    col[j]=false;
                }

        }
    }
    //将行为零 
    for(int i=0;i<matrixRowSize;i++){
        if(!row[i]){
            for(int j=0;j<matrixColSize;j++){
                matrix[i][j]=0;
            }
        }
    }
    //列变为零 
    for(int i=0;i<matrixColSize;i++){
        if(!col[i]){
            for(int j=0;j<matrixRowSize;j++){
                matrix[j][i]=0;
            }
        }
    }
//  for(int i=0;i<matrixRowSize;i++){
//      for(int j=0;j<matrixColSize;j++){
//          printf("%d  ",matrix[i][j]);
//      }
//      printf("\n");
//  }
}
//测试代码如下：
int main(void){
    int row,col;

    while(scanf("%d %d",&row,&col)!=EOF){
        int **matrix=(int **)malloc(row*sizeof(int *));
        if(matrix==NULL){
            exit(EXIT_FAILURE);
        }
        for(int i=0;i<row;i++){
            matrix[i]=(int *)malloc(col*sizeof(int));
            if(matrix[i]==NULL){
                exit(EXIT_FAILURE);
            }
        }
        for(int i=0;i<row;i++){
            for(int j=0;j<col;j++){
                scanf("%d",&matrix[i][j]);
            }
        }
        setZeroes(matrix,1,1);
    }

}

AC代码如下：