leetcode題解——832. 翻轉圖像

給定一個二進制矩陣 A,我們想先水平翻轉圖像,然後反轉圖像並返回結果。


水平翻轉圖片就是將圖片的每一行都進行翻轉,即逆序。例如,水平翻轉 [1, 1, 0] 的結果是 [0, 1, 1]。


反轉圖片的意思是圖片中的 0 全部被 1 替換, 1 全部被 0 替換。例如,反轉 [0, 1, 1] 的結果是 [1, 0, 0]。


示例 1:


輸入: [[1,1,0],[1,0,1],[0,0,0]]

輸出: [[1,0,0],[0,1,0],[1,1,1]]

解釋: 首先翻轉每一行: [[0,1,1],[1,0,1],[0,0,0]];

     然後反轉圖片: [[1,0,0],[0,1,0],[1,1,1]]



示例 2:


輸入: [[1,1,0,0],[1,0,0,1],[0,1,1,1],[1,0,1,0]]

輸出: [[1,1,0,0],[0,1,1,0],[0,0,0,1],[1,0,1,0]]

解釋: 首先翻轉每一行: [[0,0,1,1],[1,0,0,1],[1,1,1,0],[0,1,0,1]];

     然後反轉圖片: [[1,1,0,0],[0,1,1,0],[0,0,0,1],[1,0,1,0]]




說明:


    1 <= A.length = A[0].length <= 20

    0 <= A[i][j] <= 1






解題思路:

    這裏首先要做的是把一維數組逆序,然後再進行一個取反操作,因此,這裏可以藉助棧先進後出的特性來完成操作



C++

class Solution {
public:
    vector<vector<int>> flipAndInvertImage(vector<vector<int>>& A) {
        vector<vector<int>> ret;
        vector<int> tmp;
        stack<int> m_stack;
        for (int i=0; i<A.size(); i++) {
            for (int j=0; j<A[0].size(); j++) {
                m_stack.push(A[i][j]);
            }
            
            for (int j=0; j<A[0].size(); j++) {
                int val = m_stack.top();
                m_stack.pop();
                if (val == 1) val = 0;
                else val = 1;
                tmp.push_back(val);
            }
            
            ret.push_back(tmp);
            tmp.clear();
        }
        
        return ret;
    }
};



Python

class Solution:
    def flipAndInvertImage(self, A: List[List[int]]) -> List[List[int]]:
        for m_list in A:
            m_list.reverse()
        for m_list in A:
            for i in range(len(m_list)):
                m_list[i] = m_list[i] ^ 1
        return A



發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章