C++描述 LeetCode 832. 翻转图像

  大家好，我叫亓官劼（qí guān jié ）

给定一个二进制矩阵 ​

​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]]      

输入：[[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​

  ​

算法实现

class Solution {
public:
    vector<vector<int>> flipAndInvertImage(vector<vector<int>>& A) {
        int len1 = A.size();
        if(len1 == 0)
            return A;
        int len2 = A[0].size();
        int nums[2] = {1,0};
        // 水平
        for(int i = 0 ; i < len1; i++){
            int l = 0, r = len2-1;
            while(l < r){
                swap(A[i][l],A[i][r]);
                l++;
                r--;
            }
        }
        // 反转
        for(int i = 0 ; i < len1; i++){
            for(int j = 0 ; j < len2; j++){
                A[i][j] = nums[A[i][j]];
            }
        }
        return A;
    }
};