夯实算法-用栈操作构建数组

开启掘金成长之旅！这是我参与「掘金日新计划 · 2 月更文挑战」的第 24 天，点击查看活动详情

题目：LeetCode

给你一个数组 target 和一个整数 n。每次迭代，需要从  list = { 1 , 2 , 3 ..., n } 中依次读取一个数字。

请使用下述操作来构建目标数组 target ：

• "Push"：从 list 中读取一个新元素， 并将其推入数组中。
• "Pop"：删除数组中的最后一个元素。
• 如果目标数组构建完成，就停止读取更多元素。

题目数据保证目标数组严格递增，并且只包含 1 到 n 之间的数字。

请返回构建目标数组所用的操作序列。如果存在多个可行方案，返回任一即可。

示例 1：

输入： target = [1,3], n = 3
输出： ["Push","Push","Pop","Push"]
解释： 读取 1 并自动推入数组 -> [1]
读取 2 并自动推入数组，然后删除它 -> [1]
读取 3 并自动推入数组 -> [1,3]

示例 2：

输入： target = [1,2,3], n = 3
输出： ["Push","Push","Push"]

示例 3：

输入： target = [1,2], n = 4
输出： ["Push","Push"]
解释： 只需要读取前 2 个数字就可以停止。

提示：

• 1 <= target.length <= 100
• 1 <= n <= 100
• 1 <= target[i] <= n
• target 严格递增

解题思路

根据题意分析： 现在给定从1到n的n个数字，要求通过入栈，出栈操作构造出target。

• 1、从1到n依次循环执行push操作
• 2、使用二分搜索法判断当前元素是否包含在数组中
• 3、如果没有再执行一次Pop操作
• 4、停止条件使用当前元素与数组中最后一个元素比较即可

代码实现

public List < String > buildArray(int[] target, int n) {
    List < String > res = new ArrayList < > ();
    for (int i = 1; i <= n; i++) {
        res.add("Push");
        if (Arrays.binarySearch(target, i) < 0) {
            res.add("Pop");
        }
        if (i == target[target.length - 1]) {
            break;
        }
    }
    return res;
}

运行结果

复杂度分析

• 空间复杂度：$O(1)$
• 时间复杂度：$O(n)$

在掘金(JUEJIN)  一起分享知识, Keep Learning!