题目描述
请编写一个算法,若N阶方阵中某个元素为0,则将其所在的行与列清零。
给定一个N阶方阵int[]mat和矩阵的阶数n,请返回完成操作后的int[][]方阵(C++中为vector
测试样例:
[[1,2,3],[0,1,2],[0,0,1]]
返回:[[0,0,3],[0,0,0],[0,0,0]]、
solution:
# -*- coding:utf-8 -*-
class Clearer:
def clearZero(self, mat, n):
a,b = [],[]
for i in range(n):
for j in range(n):
if mat[i][j]==0:
a.append(i)
b.append(j)
for i in range(n):
for j in range(n):
if i in a or j in b:
mat[i][j]=0
return mat