题目来源:链接: [https://leetcode-cn.com/problems/flipping-an-image/submissions/].
1.问题描述
给定一个二进制矩阵 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. 1 <= A.length = A[0].length <= 20
2. 0 <= A[i][j] <= 1
2.解决方案
一开始我的代码 翻转时没有用到 reverse 函数,判断时用了两个if语句,显得有些臃肿。
最初版代码如下:
class Solution {
public:
vector<vector<int>> flipAndInvertImage(vector<vector<int>>& A) {
for(int i = 0; i < A.size(); i++)
{
int len = A[i].size() - 1;
int k = 0;
while(k < len) // 此处可以优化
{
int tmp = A[i][k];
A[i][k] = A[i][len];
A[i][len] = tmp;
k++;
len--;
}
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;
}
};
3.代码
优化代码如下:
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++)
{
A[i][j] = 1 - A[i][j];
}
}
return A;
}
};
执行用时: 24 ms, 在Flipping an Image的C++提交中击败了10.43% 的用户
内存消耗: 9.8 MB, 在Flipping an Image的C++提交中击败了0.00% 的用户
我现在严重怀疑是我的 垃圾 卡的要命的电脑限制了我的算法速度!!!哼
2019/3/2 胡云层 于南京 9