leetcode 832.翻转图像

leetcode 832.翻转图像

题干

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

题解

每行逆序后01转换，模拟即可

class Solution {
public:
    vector<vector<int>> flipAndInvertImage(vector<vector<int>>& A) {
        int rows = A.size();
        int cols = A[0].size();
        for(int i = 0 ; i < rows ; ++i){
            reverse(A[i].begin(),A[i].end());
        }
        for(int i = 0 ; i < rows ; ++i){
            for(int j = 0 ; j < cols ; ++j){
                A[i][j] = (A[i][j] + 1) % 2;
            }
        }
        return A;
    }
};

也可以直接在逆序的过程中完成01转换：
可以发现如果逆序后位置交换的两个数是不同的，那经过两个步骤后两个数相当于不变；而如果两个数相同，则两者都要进行一次01转换。

class Solution {
public:
    vector<vector<int>> flipAndInvertImage(vector<vector<int>>& A) {
        int rows = A.size();
        int cols = A[0].size();
        for(int i = 0 ; i < rows ; ++i){
            for(int j = 0 ; j <= (cols - 1) / 2 ; ++j){
                if(j == cols - j - 1){
                    A[i][j] = (A[i][j] + 1) % 2;
                }else if(A[i][j] == A[i][cols - j - 1]){
                    //要交换的两个元素相等的情况，将两者全都01翻转即可
                    A[i][j] = (A[i][j] + 1) % 2;
                    A[i][cols - j - 1] = (A[i][cols - j - 1] + 1) % 2;
                }
                //要交换的两个元素不等的时，两者都不必变化
            }
        }
        return A;
    }
};