leetcode--Set Matrix Zeroes

最新推荐文章于 2023-02-25 23:03:36 发布

原创最新推荐文章于 2023-02-25 23:03:36 发布 · 708 阅读

0 ·

CC 4.0 BY-SA版权

leetcode 同时被 2 个专栏收录

177 篇文章

订阅专栏

数组

64 篇文章

订阅专栏

本文介绍了一种优化算法，旨在通过仅使用O(m+n)的空间复杂度来实现矩阵中指定元素为零时，其所在行和列全部置零的操作，同时提供了从第一行和第一列进行标记和后续清零的详细步骤。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

Given a m x n matrix, if an element is 0, set its entire row and column to 0. Do it in place.

click to show follow up.

Follow up:

Did you use extra space?
A straight forward solution using O(mn) space is probably a bad idea.
A simple improvement uses O(m + n) space, but still not the best solution.
Could you devise a constant space solution?

题意：给定m*n的矩阵，如果某个元素是0，使其同行，同列的所有元素为0

要求使用O(m+n)空间复制度，甚至常数空间

解法1：首先遍历第一行，如果找到0,做标记，说明第一行要清0

遍历第一列，如果找到0,做标记，说明第一列要清0

然后从(1,1)开始遍历矩阵，找到0，将对应的第一行，第一列的元素设置为0，用于标记该行列应该清0

最后，根据第一行，第一列，0的情况，清0

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