刷题-力扣-832. 翻转图像

832. 翻转图像

题目链接

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/flipping-an-image/
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

题目描述

给定一个二进制矩阵 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

题目分析

1. 根据题目描述的规则，分为两步
2. 第一步反转矩阵每一行，使用双指针交换元素位置
3. 第二步将0变为1，1变为0

代码

class Solution {
public:
    vector<vector<int>> flipAndInvertImage(vector<vector<int>>& A) {
        for (int row = 0; row < A.size(); row++) {
            int left = 0;
            int right = A[row].size() - 1;
            while (left < right) {
                int temp = A[row][left];
                A[row][left] = A[row][right];
                A[row][right] = temp;
                left++;
                right--;
            }
        }
        for (int row = 0; row < A.size(); row++) {
            for (int col =0; col < A[row].size(); col++) {
                if (A[row][col] == 1) A[row][col] = 0;
                else A[row][col] = 1;
            }
        }
        return A;
    }
};