leetcode:面試題 01.08. 零矩陣(陣列,中等)

亦薇由依若菲晨發表於2020-12-27

題目:

在這裡插入圖片描述

分析:

雖然是簡單題,但還是要強調一下,
不能遇到0就修改,這樣是錯的。
可以藉助額外相等空間,但是也可以只借助一維的,直接記錄那些行,哪些列需要被設定為零,最後統一進行設定。

程式碼:

class Solution {
public:
    void setZeroes(vector<vector<int>>& matrix) {
        if(matrix.size()==0||matrix[0].size()==0) return ;
        vector<int> v1(matrix.size(),0);
        vector<int> v2(matrix[0].size(),0);
        for(int i=0;i<matrix.size();i++)
        for(int j=0;j<matrix[0].size();j++)
        if(matrix[i][j]==0) v1[i]=1,v2[j]=1;
        for(int i=0;i<matrix.size();i++)
        if(v1[i])
        for(int j=0;j<matrix[0].size();j++) matrix[i][j]=0;
        
        for(int i=0;i<matrix[0].size();i++)
        if(v2[i])
        for(int j=0;j<matrix.size();j++) matrix[j][i]=0;        
    }
};

相關文章