原题指路
翻转图像
题目描述
给定一个二进制矩阵 A
,我们想先水平翻转图像,然后反转图像并返回结果。
水平翻转图片就是将图片的每一行都进行翻转,即逆序。例如,水平翻转 [1, 1, 0]
的结果是 [0, 1, 1]
。
反转图片的意思是图片中的 0
全部被 1
替换, 1
全部被 0
替换。例如,反转 [0, 1, 1]
的结果是[1, 0, 0]
。
解题思路
这题比较简单,双指针遍历数组就行了,不过太过暴力也不好,所以需要用点小技巧。
我们发现,当前后指针所指元素不同时,翻转时不需要进行任何操作;而当前后指针所指元素相同时,我们需要将他们同时与1做异或运算。
时间复杂度:O(n)
空间复杂度:O(1)
代码
class Solution
{
public:
vector<vector<int>> flipAndInvertImage(vector<vector<int>>& A)
{
int length=A.size();
for(int i=0;i<length;i++)
for(int j=0,index=length-1;j<=index;j++,index--)//双指针遍历每一行,保证每个元素仅被访问一次
if(A[i][j]==A[i][index])//只有前后需要交换的元素相同是才需要变化
A[i][j]=A[i][index]^=1;//变化的本质其实就是将元素与1异或,这里写成这样不仅是为了方便,而且是为了防止j=index的特殊情况出错
return A;
}
};