将0所在的行列清零-二维数组和矩阵

将0所在的行列清零-二维数组和矩阵

 

public class 将零所在的行列清零 {

	public static void main(String[] args) {
		// TODO Auto-generated method stub
        int[][] arr = 
        	{{1,2,3,4},
        	 {5,6,0,8},
             {9,10,11,12},
             {13,14,15,16}};
		solve(arr);
		Print(arr);
	}

	public static void solve(int[][] arr) {
		int M = arr.length;//行宽
		int N = arr[0].length;//列宽
		//记录那些行出现了零
		int[] rowRecord = new int[M];
		//记录那些列出现了零
		int[] colRecord = new int[N];
		for(int i = 0; i<M; i++) {
			for(int j = 0; j<N; j++) {
				if(arr[i][j]==0) {
					rowRecord[i] = 1;
					colRecord[j] = 1;
				}
			}
		}
		for(int row =0; row<M; row++) {
			for(int col=0; col<N; col++) {
				if(rowRecord[row]==1||colRecord[col]==1) {
					arr[row][col] = 0;
				}
			}
		}
	}
	
	public static void Print(int[][] arr) {
		int row = arr.length;
		int col = arr[0].length ;
		for(int i = 0;i<row;i++) {
			for(int j = 0;j<col;j++) {
				System.out.print(arr[i][j] +" ");
			}
			System.out.println();
		}
	}
}

上一篇:蓝桥杯算法竞赛系列第八章——提高篇之广度优先搜索(BFS)


下一篇:BFS 的使用场景总结:层序遍历、最短路径等问题