【模拟】用栈操作构建数组

题目

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 严格递增

解题

方法一：模拟

思路

根据题意直接模拟。

具体来说：维护一个指针(idx)初始指向 target 的开头，然后枚举 $1\sim n$，每次都把枚举到的数字(i)入栈（向答案集合(ans)中追加 "Push"），如果该数字与 target[idx] 不同，说明不需要该数字，把该数字出栈（向答案集合中追加 "Pop"），如果相同，把 idx 增加向后推进直到 target 末尾停止。

最后返回 ans。

代码

class Solution {
    public List<String> buildArray(int[] target, int n) {
        List<String> ans = new ArrayList<>();
        for (int i = 1, idx = 0; i <= n && idx < target.length; ++i) {
            ans.add("Push");
            if (target[idx] == i) ++idx;
            else ans.add("Pop");
        }
        return ans;
    }
}