leetcode算法题每日一练- 翻转图像

题目

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

解析

直接按照题意,先水平翻转,也就数组倒排,再反转,就是数组值变换0变1,1变0. 所以简单暴力,借助个栈的天然先进后出的特性直接倒排,在出栈的时候进行数值反转

实现

public static int[][] flipAndInvertImage(int[][] A) {
          
   
        int length = A[0].length;
        Stack<Integer> result = new Stack<>();
        for (int i = 0; i < A.length; i++) {
          
   
            // 入栈,天然倒排
            for (int j = 0; j < length; j++) {
          
   
                result.push(A[i][j]);
            }
            // 出栈
            for (int k = 0; k < length; k++) {
          
   
                Integer pop = result.pop();
                // 值反转
                pop = pop == 1? 0 : 1;
                A[i][k] = pop;
            }
        }
        return A;
    }

总结

首先,时间复杂度嵌套了一层循环,内循环中又加一层循环。所以为O(N*N+N),空间复杂度借助栈存储所以为O(N).当然本题解法不止一种,如果读者能够观察出规律的话,就可以减少一次循环。行列规律解析可以参考官方题解

经验分享 程序员 微信小程序 职场和发展