leetcode: Set Matrix Zeroes

mmlfs

于 2014-04-12 09:26:22 发布

阅读量234

点赞数

CC 4.0 BY-SA版权

分类专栏： leetcode 文章标签： leetcode

本文链接：https://blog.youkuaiyun.com/mmlfs/article/details/23512917

leetcode 专栏收录该内容

191 篇文章

订阅专栏

本文介绍了一种高效的矩阵置零算法，通过使用两个集合记录需要置零的行和列，最终实现矩阵中特定元素所在行和列的批量置零操作。

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

用两个set记录哪些行列应该被清零，之后按照set中的记录清零即可。

public class Solution {
    public void setZeroes(int[][] matrix) {
        int m = matrix.length;
        if( m == 0 )
        {
            return;
        }
        int n = matrix[0].length;
        Set<Integer> setX = new HashSet<Integer>();
        Set<Integer> setY = new HashSet<Integer>();
        for( int i=0;i<m;i++ )
        {
            for( int j=0;j<n;j++ )
            {
                if( matrix[i][j] == 0 )
                {
                    setX.add(i);
                    setY.add(j);
                }
            }
        }
        for( int tmp : setX )
        {        
            
            for( int i=0;i<n;i++ )
            {
                matrix[tmp][i] = 0;
            }
        } 
        for( int tmp : setY )
        {        
            
            for( int i=0;i<m;i++ )
            {
                matrix[i][tmp] = 0;
            }
        } 
        return ;
    }
}