给定一个二进制矩阵 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
转载于:https://blog.51cto.com/12876518/2407605