【2022-10-15每日一题】1441. 用栈操作构建数组[Medium]
2022-10-15
2分钟阅读时长
2022-10-15每日一题:1441. 用栈操作构建数组
难度:Medium
标签:栈 、 数组 、 模拟
给你一个数组 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
严格递增
方法一:模拟
详细思路过程见官方题解,这里只做个人刷题记录,方便后续查询阅读
自己模拟写法
func buildArray(target []int, n int) []string {
ans, i := []string{}, 0
for _, x := range target {
for i++; i <= n; i++ {
// 如果它在 target 中,则只需要将它 Push 入栈即可
ans = append(ans, "Push")
// 如果不在 target 中,可以先将其 Push 入栈,紧接着 Pop 出栈
if i != x {
ans = append(ans, "Pop")
} else {
break
}
}
}
return ans
}
// 第二写法
func buildArray(target []int, n int) []string {
ans, num := []string{}, 1
for _, x := range target {
for num != x {
ans, num = append(ans, "Push", "Pop"), num+1
}
ans, num = append(ans, "Push"), num+1
}
return ans
}
// 第三写法
func buildArray(target []int, n int) []string {
ans, num := []string{}, 0
for _, x := range target {
for num++; num < x; num++ {
ans = append(ans, "Push", "Pop")
}
ans = append(ans, "Push")
}
return ans
}
官方模拟写法
思路
操作的对象是 1 到 n 按顺序排列的数字,每次操作一个数字时,如果它在 target 中,则只需要将它 Push 入栈即可。如果不在 target 中,可以先将其 Push 入栈,紧接着 Pop 出栈。因为 target 中数字是严格递增的,因此只要遍历 target,在 target 中每两个连续的数字 prev 和 number 中插入 number−prev−1 个 Push 和 Pop,再多加一个 Push 来插入当前数字即可。
代码
func buildArray(target []int, n int) (ans []string) {
prev := 0
for _, number := range target {
for i := 0; i < number-prev-1; i++ {
// 如果不在 target 中,可以先将其 Push 入栈,紧接着 Pop 出栈
ans = append(ans, "Push", "Pop")
}
ans = append(ans, "Push")
prev = number
}
return ans
}
复杂度分析
时间复杂度:O(n)。
空间复杂度:O(1)。