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-max(target)(这里称为i),设置指针index指向tartget
- 因为在最大值之前target中一定添加过i元素,只是有的删除了有的未删除,所以这里每遍历一个i值都要在答案中添加一个push。
- 判断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