清除行列 请编写一个算法,若N阶方阵中某个元素为0,则将其所在的行与列清零。 给定一个N阶方阵int[][](C++中为vector<vector><int>>)mat和矩阵的阶数n,请返回完成操

来源:互联网 发布:小孩刘心悠原帖知乎 编辑:程序博客网 时间:2024/06/06 04:20

请编写一个算法,若N阶方阵中某个元素为0,则将其所在的行与列清零。

给定一个N阶方阵int[][](C++中为vector<vector><int>>)mat和矩阵的阶数n,请返回完成操作后的int[][]方阵(C++中为vector<vector><int>>),保证n小于等于300,矩阵中的元素为int范围内。</int></vector></int></vector>

测试样例:
[[1,2,3],[0,1,2],[0,0,1]]
返回:[[0,0,3],[0,0,0],[0,0,0]]



public static int[][] clearZero(int[][] mat, int n) {
  // write code here\
  boolean hang[]=new boolean[n];
  boolean lie[]=new boolean[n];
  for(int i=0;i<mat.length;i++){
   for(int j=0;j<mat[i].length;j++){
    if(mat[i][j]==0){
     hang[i]=true;
     lie[j]=true;
    }
   }
  }
  for(int i=0;i<hang.length;i++){
   if(hang[i]){
    for(int j=0;j<lie.length;j++){
     mat[i][j]=0;
    }
   }
   if(lie[i]){
    for(int j=0;j<lie.length;j++){
     mat[j][i]=0;
    }
   }
  }
  return mat;
 }

阅读全文
0 0