LeetCode

832. 翻转图像

2018-11-05  本文已影响2人  闭门造折

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

思路

其实就是同行对称交换的同时,反转结果
两个for循环,一个用来遍历所有行
每一行再遍历每一项,只需要遍历到中间元素就可以了,否则会出现重复交换的问题

具体代码

class Solution {
    public void swap(int[][] A, int x, int y1, int y2){
        if(y1 == y2){ //如果是中间元素,就只反转就行
            A[x][y1] = 1 - A[x][y1];
            return;
        }
        //非中间元素,边交换边反转
        int t = 1 - A[x][y1];
        A[x][y1] = 1 - A[x][y2];
        A[x][y2] = t;
        return;
    }
    public int[][] flipAndInvertImage(int[][] A) {
        int len = A.length;
        for(int i = 0; i < len; i++){ //遍历所有行
            for(int j = 0; j <= (len - 1) / 2; j++){ //每行从两侧向中间交换
                swap(A, i, j, len - j - 1); //调用交换函数
            }
        }
        return A;
    }
}
上一篇下一篇

猜你喜欢

热点阅读