LeetCode每日一题: 反转图像(No.832)

590 阅读2分钟

题目:反转图像


  给定一个二进制矩阵 A,我们想先水平翻转图像,然后反转图像并返回结果。
  水平翻转图片就是将图片的每一行都进行翻转,即逆序。例如,水平翻转 [1, 1, 0] 的结果是 [0, 1, 1]。
  反转图片的意思是图片中的 0 全部被 1 替换, 1 全部被 0 替换。例如,反转 [0, 1, 1] 的结果是 [1, 0, 0]。

示例:


输入: [[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与每个元素相减即可。
但是稍微思考后会发现:
将要翻转的两个数比较,如果两数相等即都为0或者都为1,水平反转之后其实并无变化,再反转图片操作后原来0变为1,1变为0.
如果两数不等,则水平反转后位置调换,但是反转图片后又会变回原数,相当于没有操作。
所以可以得到结论:比较水平翻转的两个数若不等则无需操作,若相等则直接用1与其相减即可。
注意一下如果一行元素是奇数,中间那个元素要记得处理一下。

实现:


  class Solution {
    public int[][] flipAndInvertImage(int[][] A) {
        for (int[] a : A) {
            for (int count = 0; count < a.length / 2; count++) {
                if (a[count] == a[a.length - 1 - count]) {
                    a[count] = 1 - a[count];
                    a[a.length - 1 - count] = 1 - a[a.length - 1 - count];
                }
            }
            if (a.length % 2 == 1) {
                a[a.length / 2] = 1 - a[a.length / 2];
            }
        }
        return A;
    }
}