Leetcode学习笔记-946-验证栈序列

2018-12-01 阅读量

题目：
给定 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 的排列。

解题思路：

判断合不合法，用个栈试一试:
把压栈的元素按顺序压入，当栈顶元素和出栈的第一个元素相同，则将该元素弹出，出栈列表指针后移并继续判断。最后判断出栈列表指针是否指向出栈列表的末尾即可。

Java：

public boolean validateStackSequences(int[] pushed, int[] popped) {

    Stack<Integer> stack = new Stack();
    int j = 0;
    for (int elem : pushed) {
        stack.push(elem);
        while (j < popped.length && !stack.isEmpty() && stack.peek() == popped[j]) {
            stack.pop();
            j++;
        }
    }
    return j == popped.length;
}

Python：

def validateStackSequences(self, pushed, popped):
    """
    :type pushed: List[int]
    :type popped: List[int]
    :rtype: bool
    """
    stack, j, elemsNum = [], 0, len(pushed)
    for elem in pushed:
        stack.append(elem)
        while j < elemsNum and len(stack) > 0 and stack[-1] == popped[j]:
            stack.pop(-1)
            j += 1
    return j == elemsNum