LeetCode总结——题832

题目来源：https://leetcode-cn.com/

 

---

 

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

思路：

对矩阵的每一行双向遍历。
遍历的当前行左右两端元素相同的取反，不同不变。

代码：

class Solution {
    public int[][] flipAndInvertImage(int[][] A) {
         for (int i = 0; i < A.length; i++) {
            for (int j = 0, k = A[i].length-1; j <= k; j++, k--) {
                if (j == k) {
                    A[i][j] = 1 - A[i][j];
                    break;
                }
                if (A[i][j] == A[i][k]) {
                    A[i][j] = 1 - A[i][k];
                    A[i][k] = A[i][j];
                }
                
            }
        }
        return A;
    }
}