leetcode_946. 验证栈序列

验证栈序列

题目

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

示例 1:

输入: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:

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

提示:

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元素

解答

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%的用户

分享: