题目
给定一个二进制矩阵 A,我们想先水平翻转图像,然后反转图像并返回结果。
水平翻转图片就是将图片的每一行都进行翻转,即逆序。例如,水平翻转 [1, 1, 0] 的结果是 [0, 1, 1]。
反转图片的意思是图片中的 0 全部被 1 替换, 1 全部被 0 替换。例如,反转 [0, 1, 1] 的结果是 [1, 0, 0]。
示例
输入:[[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]]
输入:[[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
解题思路:
假设矩阵的行数和列数都是 n,考虑列下标 left 和 right,其中 left<right 且 left+right=n−1,当 0≤i<n 时,对第 i 行进行水平翻转之后,A[i][left] 和 A[i][right] 的元素值会互换,进行反转之后,A[i][left] 和 A[i][right] 的元素值都会改变。
我们可以发现对于[1,0,0]输出的是[1,1,0],即首尾1和0对应的时候,值不相等,在翻转再反转之后,其值仍和初始值一样,无变动。这意味着当 A[i][left] 和 A[i][right] 不等的时候,其值不需要变动。那么,考虑另一种情况,即 A[i][left] 和 A[i][right] 相等的时候,比如[1,0,0,1]输入,输出之后为[0,1,1,0],可以发现在对应相等的时候其值改变,此时我们可以将其与1做异或处理。
需要注意的是,当n为奇数的时候我们要单独拿出来考虑一下,所以见下方:
class Solution {
public:
vector<vector<int>> flipAndInvertImage(vector<vector<int>>& A) {
int n=A.size();
for(int i=0;i<n;i++){
int left=0,right=n-1;
while(left<right){
if(A[i][left]==A[i][right]){
A[i][left]^=1;
A[i][right]^=1;
}
left++;
right--;
}
if(left==right){
A[i][left]^=1;
}
}
return A;
}
};
python语言:
class Solution:
def flipAndInvertImage(self, A: List[List[int]]) -> List[List[int]]:
n = len(A)
for i in range(n):
left, right = 0, n - 1
while left < right:
if A[i][left] == A[i][right]:
A[i][left] ^= 1
A[i][right] ^= 1
left += 1
right -= 1
if left == right:
A[i][left] ^= 1
return A
题目来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/flipping-an-image/solution/fan-zhuan-tu-xiang-by-leetcode-solution-yljd/