Swift-数组清零

题目:如果数组M*N中某个元素为0,将其所在的行与列清零.

题目很简单,注意很容易出现,数组所有的数据最终都是零,设置为0之前需要记录行列.

核心代码:

func clearZero(data:inout [[Int]])  {
        
        var rows:[Bool] = [Bool].init(repeating: false, count: data.count)
        var cols:[Bool] = [Bool].init(repeating: false, count: data[0].count)
        
        for i in 0..<rows.count {
            for j in 0..<cols.count {
                if data[i][j] == 0 {
                    rows[i] = true
                    cols[j] = true
                }
            }
        }
        
        for i in 0..<rows.count {
            for j in 0..<cols.count {
                if rows[i] || cols[j] {
                    data[i][j] = 0
                }
            }
        }
    }

测试代码:

var clearData:[[Int]] = [[1,2,3,4,0],[5,6,7,8,9],[9,10,11,12,1],[13,14,15,16,17]]
myString.clearZero(data: &clearData)
print("FlyElephant--数组清零--\(clearData)")
最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

推荐阅读更多精彩内容