給定乙個 m x n 的矩陣,如果乙個元素為 0,則將其所在行和列的所有元素都設為 0。請使用原地演算法。
示例 1:
輸入:
[ [1,1,1],
[1,0,1],
[1,1,1]
]輸出:
[ [1,0,1],
[0,0,0],
[1,0,1]
]
空間複雜度為o(mn)的解法自不用多說,直接新建乙個和matrix等大小的矩陣,然後一行一行的掃,只要有0,就將新建的矩陣的對應行全賦0,行掃完再掃列,然後把更新完的矩陣賦給matrix即可,這個演算法的空間複雜度太高。將其優化到o(m+n)的方法是,用乙個長度為m的一維陣列記錄各行中是否有0,用乙個長度為n的一維陣列記錄各列中是否有0,最後直接更新matrix陣列即可。這道題的要求是用o(1)的空間,那麼我們就不能新建陣列,我們考慮就用原陣列的第一行第一列來記錄各行各列是否有0.
- 先掃瞄第一行第一列,如果有0,則將各自的flag設定為true
- 然後掃瞄除去第一行第一列的整個陣列,如果有0,則將對應的第一行和第一列的數字賦0
- 再次遍歷除去第一行第一列的整個陣列,如果對應的第一行和第一列的數字有乙個為0,則將當前值賦0
- 最後根據第一行第一列的flag來更新第一行第一列
public void setzeroes(int matrix)
}for(int i = 0; i < n; i++)
}for(int i = 0; i < m; i++)}}
for(int i = 1; i < m; i++)}}
if(rowzero)
}if(clozero)
}}
參考: leetcode 73 矩陣置零
給定乙個 m x n 的矩陣,如果乙個元素為 0,則將其所在行和列的所有元素都設為 0。請使用原地演算法。示例 1 輸入 1,1,1 1,0,1 1,1,1 輸出 1,0,1 0,0,0 1,0,1 示例 2 輸入 0,1,2,0 3,4,5,2 1,3,1,5 輸出 0,0,0,0 0,4,5,0...
leetcode 73 矩陣置零
給定乙個 m x n 的矩陣,如果乙個元素為 0,則將其所在行和列的所有元素都設為 0。請使用原地演算法。示例 1 輸入 1,1,1 1,0,1 1,1,1 輸出 1,0,1 0,0,0 1,0,1 示例 2 輸入 0,1,2,0 3,4,5,2 1,3,1,5 輸出 0,0,0,0 0,4,5,0...
LeetCode 73 矩陣置零
給定乙個 m x n 的矩陣,如果乙個元素為 0,則將其所在行和列的所有元素都設為 0。請使用原地演算法。示例 1 輸入 1,1,1 1,0,1 1,1,1 輸出 1,0,1 0,0,0 1,0,1 示例 2 輸入 0,1,2,0 3,4,5,2 1,3,1,5 輸出 0,0,0,0 0,4,5,0...