Given a binary matrix A
, we want to flip the image horizontally, then invert it, and return the resulting image.
To flip an image horizontally means that each row of the image is reversed. For example, flipping [1, 1, 0]
horizontally results in [0, 1, 1]
.
To invert an image means that each 0
is replaced by 1
, and each 1
is replaced by 0
. For example, inverting [0, 1, 1]
results in [1, 0, 0]
.
Example 1:
Input: [[1,1,0],[1,0,1],[0,0,0]] Output: [[1,0,0],[0,1,0],[1,1,1]] Explanation: First reverse each row: [[0,1,1],[1,0,1],[0,0,0]]. Then, invert the image: [[1,0,0],[0,1,0],[1,1,1]]
Example 2:
Input: [[1,1,0,0],[1,0,0,1],[0,1,1,1],[1,0,1,0]] Output: [[1,1,0,0],[0,1,1,0],[0,0,0,1],[1,0,1,0]] Explanation: First reverse each row: [[0,0,1,1],[1,0,0,1],[1,1,1,0],[0,1,0,1]]. Then invert the image: [[1,1,0,0],[0,1,1,0],[0,0,0,1],[1,0,1,0]]
#pragma once
#include<iostream>
#include<vector>
using namespace std;
class Solution {
public:
vector<vector<int>> flipAndInvertImage(vector<vector<int>>& A) {
for (int i = 0; i < A.size(); i++)
{
int first =0, end = A[i].size()-1; //一个头,一个尾
for (int j = 0; j < A[i].size(); j++)
{
if (first >end) //后面的大于前面的
{
//前后交换
int temp = A[i][first];
A[i][first] = abs(A[i][end] );
A[i][end] = temp;
//取反
A[i][first] = abs(A[i][first] -1);
A[i][end] = abs(A[i][end] - 1);
}
else if (first == end) //相等的时候
{
A[i][first]=abs(A[i][first] - 1);
}
first++;
end--;
}
}
return A;
}
};
二维vector初始化
int a[2][2] = { {1,1},{1,1}};
vector<vector<int>> vect;
for (int i = 0; i < 2; i++)
{
vector<int> t(a[i],a[i] + 2);
vect.push_back(t);
}