【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":删除数组中的最后一个元素。
  • 如果目标数组构建完成,就停止读取更多元素。

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

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

 

示例 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)。

LeetCode题库地址

推荐阅读题解