编写一种算法,若M × N矩阵中某个元素为0,则将其所在的行与列清零。例:
输入:
1 1 1
1 0 1
1 1 1
输出:
1 0 1
0 0 0
1 0 1
我的思路是先将数组中的所有 0 找出,并且通过使用矩阵本身来存储需要清零的行与列的数据。
#include<stdio.h>
void setZeroes(int** matrix, int line, int column){
int i,j,x;
for(i=0;i<line;i++)
{
for(j=0;j<column;j++)
{
if(matrix[i][j]==0)
matrix[i][j]='-1'; //找到 0 并把它换成 ‘-1’ (字符,因为数组可能存在整型 -1)
}
}
for(i=0;i<line;i++)
{
for(j=0;j<column;j++)
{
if(matrix[i][j]=='-1') //找到数组中的 ‘-1’
{
x=0;
while(x<column)
{
if(matrix[i][x]!='-1') //把该行除 ‘-1’ 外的所有数换成 0
matrix[i][x]=0;
x++;
}
x=0;
while(x<line)
{
if(matrix[x][j]!='-1') //把该列除 ‘-1’ 外的所有数换成 0
matrix[x][j]=0;
x++;
}
}
}
}
for(i=0;i<line;i++)
{
for(j=0;j<column;j++)
{
if(matrix[i][j]=='-1') //把数组中的所有 ‘-1’ 换成 0(前面不将 ‘-1’ 换成 0,是防止该行或列中的其他 ‘-1’ 忽略掉,导致错误)
matrix[i][j]=0;
}
}
}
int main()
{
int* p[3];
int matrix[3][3]={1,1,1,
1,0,1,
1,1,1};
p[0]=&matrix[0][0];
p[1]=matrix[1];
p[2]=matrix[2];
setZeroes(p,3,3);
for(int i=0;i<3;i++)
{
for(int j=0;j<3;j++)
{
printf("%d ",matrix[i][j]);
}
printf("\n");
}
return 0;
}