题目描述:
给定一个二进制矩阵 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
相关话题:数组
解题思路:
将数组每一行翻转;
从头到尾遍历该数组;
如果某个元素的值为0,将其置为1,否则,将其置为0。
class Solution {
public:
vector<vector<int>> flipAndInvertImage(vector<vector<int>>& A) {
for(int i = 0; i < A.size(); ++i){
reverse(A[i].begin(), A[i].end());
for(int j = 0; j < A[i].size(); ++j){
if(A[i][j] == 0){
A[i][j] = 1;
} else {
A[i][j] = 0;
}
}
}
return A;
}
};
/**
* Return an array of arrays of size *returnSize.
* The sizes of the arrays are returned as *columnSizes array.
* Note: Both returned array and *columnSizes array must be malloced, assume caller calls free().
*/
int** flipAndInvertImage(int** A, int ARowSize, int *AColSizes, int** columnSizes, int* returnSize) {
for(int i = 0; i < ARowSize; ++i){
for(int j = 0; j < (A[i].size())/2; ++j){
int temp = A[i][j];
A[i][j] = A[i][A[i].size() - j - 1];
A[i][A[i].size() - j - 1] = temp;
}
for(int j = 0; j < A[i].size(); ++j){
if(A[i][j] == 0){
A[i][j] = 1;
} else{
A[i][j] = 0;
}
}
}
return * returnSize;
}
技术要点:数组翻转,可以使用STL中的翻转函数,也可以将矩阵中的每一行中的元素翻转(以数组元素中间的元素为轴,将两边的元素进行翻转)。
总结:对于翻转类问题,可以自己编写翻转函数来翻转元素,也可以使用STL中提供的方法进行翻转。个人感觉,在联系的时候多自己编写,在做项目的时候多使用STL中提供的函数,以提高开发效率。