【LeetCode刷题(中等程度)】剑指 Offer 31. 栈的压入、弹出序列

输入两个整数序列,第一个序列表示栈的压入顺序,请判断第二个序列是否为该栈的弹出顺序。假设压入栈的所有数字均不相等。例如,序列 {1,2,3,4,5} 是某栈的压栈序列,序列 {4,5,3,2,1} 是该压栈序列对应的一个弹出序列,但 {4,3,5,1,2} 就不可能是该压栈序列的弹出序列。

示例 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 之前弹出。

来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/zhan-de-ya-ru-dan-chu-xu-lie-lcof
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。

思路:本题的实质是判断序列是不是栈混洗序列,O(n)复杂度,直接借助一个辅助栈,模拟栈混洗的过程,每次temp.pop()之前,检测temp是否为空,或需弹出的元素在temp中,但却不是栈顶元素,出现这种情况则不是栈混洗序列。

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

代码:

class Solution {
public:
    bool validateStackSequences(vector<int>& pushed, vector<int>& popped) {
        int start = 0;
        int end = popped.size();
        stack<int> temp;
        for(const auto & c : pushed)
        {
            temp.push(c);
            while(start < end && !temp.empty()&&(popped[start] == temp.top()))
            {
                start++;
                temp.pop();
            }
        }
        return start == end;
    }
    
};

猜你喜欢

转载自blog.csdn.net/qq_33197518/article/details/108804106
今日推荐