LeetCode 1441. 用栈操作构建数组

1441. 用栈操作构建数组

给你一个数组 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

Use “Push” for numbers to be kept in target array and [“Push”, “Pop”] for numbers to be discarded.

解法1：模拟

栈的特点是 后进先出LIFO / 先进后出 FILO.

思路

操作的对象是 1 到 n 按顺序排列的数字，每次操作一个数字时，如果它在 target 中，则只需要将它 Push 入栈即可。如果不在 target 中，可以先将其 Push 入栈，紧接着 Pop 出栈。因为 target 中数字是严格递增的，因此只要遍历 target，在 target 中每两个连续的数字 prev 和 number 中插入 number−prev−1 个 Push 和 Pop，再多加一个 Push 来插入当前数字即可。

Java版：

class Solution {
    public List<String> buildArray(int[] target, int n) {
        List<String> ans = new ArrayList<>();
        int prev = 0;
        for (int number: target) {
            for (int i = 0; i < number - prev - 1; i++) {
                ans.add("Push");
                ans.add("Pop");
            }
            ans.add("Push");
            prev = number;
        }
        return ans;
    }
}

Python3版：

class Solution:
    def buildArray(self, target: List[int], n: int) -> List[str]:
        ans = []
        prev = 0
        for number in target:
            for _ in range(number - prev - 1):
                ans.append("Push")
                ans.append("Pop")
            ans.append("Push")
            prev = number
        return ans

复杂度分析

• 时间复杂度：O(n)。Push 需要添加 O(n) 次。

• 空间复杂度：O(1)。除了保存结果的数组，其他只消耗常数空间。