LeetCode每日一题——1441. 用栈操作构建数组

LeetCode每日一题系列

题目：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遍历到target中最大值即可找到所有情况。

1. 遍历1-max(target)(这里称为i)，设置指针index指向tartget
2. 因为在最大值之前target中一定添加过i元素，只是有的删除了有的未删除，所以这里每遍历一个i值都要在答案中添加一个push。
3. 判断target[index]是否等于i，如果等于i，说明该元素保留了下来，将index后移一位即可跳过。如果target[index]不等于i，说明该元素被删除，在结果中添加pop，并且index下标不变继续向下比较。

题解

class Solution:
    def buildArray(self, target: List[int], n: int) -> List[str]:
        # 结果数组
        ans = []
        # 指针
        index = 0
        # 遍历1-max(target)
        for i in range(1, max(target)+1):
            # 每个i值都添加push
            ans.append('Push')
            # 被删除元素的情况
            if i != target[index]:
                ans.append('Pop')
            # 被保留元素的情况
            else:
                index += 1
        return ans