Write an algorithm such that if an element in an MxN matrix is 0, its entire row and column is set to 0
写一个算法使得把一个M*N的矩阵中0元素所在位置的行列设置为零
At first glance, this problem seems easy: just iterate through the matrix and every time we see a 0, set that row and column to 0 There’s one problem with that solution though: we will “recognize” those 0s later on in our iteration and then set their row and column to zero Pretty soon, our entire matrix is 0s!
One way around this is to keep a second matrix which flags the 0 locations We would then do a second pass through the matrix to set the zeros This would take O(MN) space.Do we really need O(MN) space? No Since we’re going to set the entire row and column to zero, do we really need to track which cell in a row is zero? No We only need to know that row 2, for example, has a zero. The code below implement this algorithm We keep track in two arrays all the rows with zeros and all the columns with zeros We then make a second pass of the matrix and set a cell to zero if its row or column is zerovoid setZeros(int matrix[][],int m,int n){ int *row=(int *)malloc(m*sizeof(int)); int *column=(int *)malloc(n*sizeof(int)); int i,j; for(i=0;i