给定一个二进制矩阵 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
解答
遍历:
class Solution {
public:
vector<vector<int>> flipAndInvertImage(vector<vector<int>>& A) {
for(auto& row : A){
reverse(row.begin(), row.end());
}
for(auto& row : A){
for(auto& elememt : row){
elememt ^= 1;
}
}
return A;
}
};
参考评论区,实际上只需要遍历一半元素即可,如果对称位置的元素不相等,翻转后再进行替换,实际上等于没有进行任何操作,因此只需要对对称位置元素相等的情况进行处理。:
class Solution {
public:
vector<vector<int>> flipAndInvertImage(vector<vector<int>>& A) {
int rows = A.size();
int cols = A[0].size();
for(int i = 0; i < rows; i++){
for(int j = 0; j < ((cols + 1) >> 1); j++){
if(A[i][j] == A[i][cols - j - 1]){
int temp = 1 - A[i][j];
A[i][j] = A[i][cols - j - 1] = temp;
}
}
}
return A;
}
};