【leetcode】73.(Medium)Set Matrix Zeroes

解题思路:
首先空间复杂度为O(m*n)的做法是,直接copy一套matrix来记录
空间复杂度为O(m+n)的做法是设置数组row[m]和column[n]来记录每行每列的0的情况
空间复杂度为O(1)的情况就是设置两个boolean变量,首先判定第一行和第一列是否会最终变为0,然后从第一行第一列(最左上角位置为第0行第0列)开始,更新matrix。最后再根据这两个boolean变量来更新matrix的第一行和第一列。

提交代码:

class Solution {
    public void setZeroes(int[][] matrix) {
    	boolean zeroColumn=false,zeroRow=false;
    	
    	for(int i=0;i<matrix.length;i++)
        	if(matrix[i][0]==0) {
        		zeroColumn=true;
        		break;
        	}
    	
    	for(int j=0;j<matrix[0].length;j++)
    		if(matrix[0][j]==0) {
        		zeroRow=true;
        		break;
        	}
    	
    	for(int i=0;i<matrix.length;i++)
    		for(int j=0;j<matrix[0].length;j++)
    			if(matrix[i][j]==0) {
    				matrix[i][0]=0;
    				matrix[0][j]=0;
    			}
    			
    	for(int i=1;i<matrix.length;i++)
    		for(int j=1;j<matrix[0].length;j++)
    			if(matrix[i][0]==0||matrix[0][j]==0)
    				matrix[i][j]=0;
    	
    	if(zeroColumn)
    		for(int i=0;i<matrix.length;i++)
    			matrix[i][0]=0;
    	if(zeroRow)
    		for(int j=0;j<matrix[0].length;j++)
    			matrix[0][j]=0;
    }
}

运行结果:
在这里插入图片描述

猜你喜欢

转载自blog.csdn.net/AXIMI/article/details/84319346