若一个M*N的举证当中某个元素为零，则将其所有的行和列清零。

最新推荐文章于 2022-07-29 19:58:14 发布

dongchen5700

最新推荐文章于 2022-07-29 19:58:14 发布

阅读量160

点赞数

文章标签：面试 java 数据结构与算法

原文链接：http://www.cnblogs.com/ProWhalen/p/5362382.html

版权

这是一道程序员面试金典上面的练习题；读完题其实结题思路很明显了；

输入矩阵—>搜索出为零的部位—>利用为零的部位信息重新遍历矩阵给该赋值0的位点给0—>结束

package shounue;
import java.util.Scanner;
public class SetMatrixZeroes {
public static void setZeros(int[][] matrix){
    boolean [] row = new boolean[matrix.length];
    boolean [] column = new boolean[matrix[0].length];
    for(int i = 0; i < matrix.length; i++)
        for(int j = 0; j < matrix[0].length; j++)
        {
            if(matrix[i][j] == 0) 
            {
                row[i] = true;
                column[j] = true;
            }
        }
    for(int i = 0; i < matrix.length; i++)
        for(int j = 0; j < matrix[0].length;j++)
        {
            if(row[i]||column[j])
                matrix[i][j] = 0;
        }
}    
public static void main(String [] args){
    Scanner in = new Scanner(System.in);
    System.out.println("please input row and column");
    int row = in.nextInt();
    int column = in.nextInt();
    int [][] Arrays = new int[row][column];
    for(int i = 0; i < Arrays.length; i++)
       for(int j = 0; j < Arrays[0].length;j++)
             Arrays[i][j] = in.nextInt();    
    setZeros(Arrays);
    for(int i = 0; i < Arrays.length; i++)
    { for(int j = 0; j < Arrays[0].length;j++)
              { System.out.print(Arrays[i][j]);
                System.out.print(" ");
              }
      System.out.println();
    }
}
}

可以看到，常规解法时间复杂度还是蛮大的需要两次O（N*M）的统计过程；

回想到Leetcode 有相同的题目：73. Set Matrix Zeroes

解法如下：

public class Solution {
    public void setZeroes(int[][] matrix) {
        if(matrix == null || matrix.length == 0 || matrix[0].length == 0) return; //好的算法，都会事先对不可能的情况进行处理；数组处理NULL 和值
        int m = matrix.length;
        int n = matrix[0].length;
        boolean row = false;
        boolean col = false;
        for(int i = 0; i < m; i++)
          {
              for(int j = 0; j < n; j++)
                {
                    if(matrix[i][j]==0) 
                     {
                         matrix[0][j] = 0;
                         matrix[i][0] = 0;
                         if(i==0) row = true;
                         if(j==0) col = true;
                     }
                }
          }
         for(int i = 1; i < m; i++)
         {
             if(matrix[i][0] == 0) {
                 for(int j = 1; j < n; j++)  matrix[i][j] = 0;
             }
         }
         for(int j = 1; j < n; j++)
         {
             if(matrix[0][j] == 0){
                 for(int i = 1; i < m; i++) matrix[i][j] = 0;
             }
         }
         if (row){
             for(int j = 0; j < n; j++) matrix[0][j] = 0;
         }
         if(col){
             for(int i = 0; i < m; i++) matrix[i][0] = 0;
         }
    }
}