LC 1441. 用栈操作构建数组

题目描述

这是 LeetCode 上的 1441. 用栈操作构建数组 ,难度为 中等

给你一个数组 target 和一个整数 n

每次迭代,需要从 list = { 1 , 2 , 3 ..., n } 中依次读取一个数字。

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

  • "Push":从 list 中读取一个新元素, 并将其推入数组中。
  • "Pop":删除数组中的最后一个元素。

如果目标数组构建完成,就停止读取更多元素。

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

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

示例 1:

1
2
3
4
5
6
7
8
输入:target = [1,3], n = 3

输出:["Push","Push","Pop","Push"]

解释:
读取 1 并自动推入数组 -> [1]
读取 2 并自动推入数组,然后删除它 -> [1]
读取 3 并自动推入数组 -> [1,3]

示例 2:
1
2
3
输入:target = [1,2,3], n = 3

输出:["Push","Push","Push"]

示例 3:
1
2
3
4
5
输入:target = [1,2], n = 4

输出:["Push","Push"]

解释:只需要读取前 2 个数字就可以停止。

提示:

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

模拟

根据题意进行模拟即可。

每次我们将当前处理到 i 压入栈中(往答案添加一个 Push),然后判断当前处理到的 i 是否最新的栈顶元素 $target[j]$ 是否相同。

若不相同则丢弃元素(往答案添加一个 Pop),若存在则将指针 j 后移,直到构建出目标答案。

Java 代码:

1
2
3
4
5
6
7
8
9
10
11
12
class Solution {
public List<String> buildArray(int[] target, int n) {
List<String> ans = new ArrayList<>();
int m = target.length;
for (int i = 1, j = 0; i <= n && j < m; i++) {
ans.add("Push");
if (target[j] != i) ans.add("Pop");
else j++;
}
return ans;
}
}

C++ 代码:
1
2
3
4
5
6
7
8
9
10
11
12
13
class Solution {
public:
vector<string> buildArray(vector<int>& target, int n) {
int m = target.size();
vector<string> ans;
for (int i = 1, j = 0; i <= n && j < m; i++) {
ans.push_back("Push");
if (target[j] != i) ans.push_back("Pop");
else j++;
}
return ans;
}
};

Python3 代码:
1
2
3
4
5
6
7
8
9
10
11
12
class Solution:
def buildArray(self, target: List[int], n: int) -> List[str]:
ans = []
m, i, j = len(target), 1, 0
while i <= n and j < m:
ans.append("Push")
if target[j] != i:
ans.append("Pop")
else:
j += 1
i += 1
return ans

TypeScript 代码:
1
2
3
4
5
6
7
8
9
10
function buildArray(target: number[], n: number): string[] {
const ans = new Array<string>()
const m = target.length
for (let i = 1, j = 0; i <= n && j < m; i++) {
ans.push("Push")
if (target[j] != i) ans.push("Pop")
else j++
}
return ans
}

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

最后

这是我们「刷穿 LeetCode」系列文章的第 No.1441 篇,系列开始于 2021/01/01,截止于起始日 LeetCode 上共有 1916 道题目,部分是有锁题,我们将先把所有不带锁的题目刷完。

在这个系列文章里面,除了讲解解题思路以外,还会尽可能给出最为简洁的代码。如果涉及通解还会相应的代码模板。

为了方便各位同学能够电脑上进行调试和提交代码,我建立了相关的仓库:https://github.com/SharingSource/LogicStack-LeetCode

在仓库地址里,你可以看到系列文章的题解链接、系列文章的相应代码、LeetCode 原题链接和其他优选题解。


本博客所有文章除特别声明外,均采用 CC BY-SA 4.0 协议 ,转载请注明出处!