验证栈序列

题目

给定 pushed 和 popped 两个序列，每个序列中的 值都不重复，只有当它们可能是在最初空栈上进行的推入 push 和弹出 pop 操作序列的结果时，返回 true；否则，返回 false 。

示例 1：

1
2
3
4
5

输入：pushed = [1,2,3,4,5], popped = [4,5,3,2,1]
输出：true
解释：我们可以按以下顺序执行：
push(1), push(2), push(3), push(4), pop() -> 4,
push(5), pop() -> 5, pop() -> 3, pop() -> 2, pop() -> 1

示例 2：

1
2
3

输入：pushed = [1,2,3,4,5], popped = [4,3,5,1,2]
输出：false
解释：1 不能在 2 之前弹出。

提示：

1
2
3

0 <= pushed.length == popped.length <= 1000
0 <= pushed[i], popped[i] < 1000
pushed 是 popped 的排列。

来源：力扣（LeetCode） 链接：https://leetcode-cn.com/problems/validate-stack-sequences 著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

思路

创建辅助栈sub，扫描两个序列

  • 遍历出栈pop序列
    • 如果sub栈顶不等于pop[i]
      • 如果i到了末尾， return false
      • push序列取元素入sub栈
      • 直到匹配
    • sub出栈，继续判断下一个pop元素

解答

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31

func validateStackSequences(pushed []int, popped []int) bool {
    var (
        stk = make([]int, 0, len(pushed))  // 辅助栈
        ps int  // 入栈序列的指针
    )

    for _, v := range popped {
        // 栈空
        if len(stk) == 0 {
            stk = append(stk, pushed[ps])
            ps++
        }

        for v != stk[len(stk)-1] {
            // 入栈序列用完了，出栈序列还有未匹配的
            if ps == len(pushed) {
                return false
            }
            // 入栈序列推一个到辅助栈
            stk = append(stk, pushed[ps])
            ps++
        }
        // 出现匹配，辅助栈出栈
        stk = stk[:len(stk)-1]
    }
    // 整个出栈序列检测完
    return true
}

// 执行用时 :8 ms, 在所有 Go 提交中击败了90.28%的用户
// 内存消耗 :3.8 MB, 在所有 Go 提交中击败了100.00%的用户