矩阵置零Python解法
生活随笔
收集整理的這篇文章主要介紹了
矩阵置零Python解法
小編覺得挺不錯的,現在分享給大家,幫大家做個參考.
給定一個?m?x?n?的矩陣,如果一個元素為?0?,則將其所在行和列的所有元素都設為?0?。請使用?原地?算法。
例:
輸入:matrix = [[1,1,1],[1,0,1],[1,1,1]] 輸出:[[1,0,1],[0,0,0],[1,0,1]]解析:遍歷一遍,將為0的位置進行存儲,然后遍歷存儲的值進行置0即可。
class Solution(object):def setZeroes(self, matrix):""":type matrix: List[List[int]]:rtype: None Do not return anything, modify matrix in-place instead."""m = len(matrix) # 行數n = len(matrix[0]) # 列數ans = [] # 初始化存儲列表for i in range(m): # 遍歷找0for j in range(n):if matrix[i][j] == 0:ans.append([i, j]) # 存儲for i in ans:for j in range(m):matrix[j][i[1]] = 0 # 列置0matrix[i[0]][:]=[0]*n # 行置0return matrix 創作挑戰賽新人創作獎勵來咯,堅持創作打卡瓜分現金大獎總結
以上是生活随笔為你收集整理的矩阵置零Python解法的全部內容,希望文章能夠幫你解決所遇到的問題。
- 上一篇: 新魔法药水
- 下一篇: 删除有序数组中的重复项 IIPython