給定一個二進制矩陣 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