栈的压入、弹出序列

问题描述

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

来源：牛客网
链接：https://www.nowcoder.com/practice/d77d11405cc7470d82554cb392585106?tpId=13&tqId=11174&tPage=2&rp=2&ru=/ta/coding-interviews&qru=/ta/coding-interviews/question-ranking

C++代码实现

思路：使用一个栈，同时遍历压入序列和弹出序列，让栈中元素和弹出序列比较，当栈为空时，将压入序列的值压入栈，每次判断栈顶元素是否和弹出序列值相同，相同则栈弹出值，并使弹出序列位置后移，否则将压入序列的值压入栈，直到压入序列遍历完毕，再依次弹出栈中元素与弹出序列相比较。
不知道为什么这里c++的数组没有越界问题…

class Solution {
public:
    bool IsPopOrder(vector<int> pushV,vector<int> popV) {
        stack<int> s;
        int i = 0, j = 0;
        while(i < pushV.size())
        {
            if (s.empty())
            {
                s.push(pushV[i++]);
            }
            if (s.top() == popV[j])
            {
                s.pop();
                j++;
            }
            else
            {
                s.push(pushV[i++]);
            }
        }
        while(!s.empty())
        {
            if (s.top() != popV[j])
                return false;
            s.pop();
            j++;
        }
        return true;
    }
};

python2代码实现

python没有栈，使用列表

# -*- coding:utf-8 -*-
class Solution:
    def IsPopOrder(self, pushV, popV):
        # write code here
        i = 0
        j = 0
        s = []
        while i < len(pushV):
            if s == []:
                s.append(pushV[i])
                i += 1
            if s[len(s)-1] == popV[j]:
                s.pop()
                j += 1
            else:
                if i < len(pushV):
                    s.append(pushV[i])
                    i += 1
        while s != []:
            if s.pop() != popV.pop(j):
                return False
        return True

java代码实现

import java.util.Stack;
public class Solution {
    public boolean IsPopOrder(int [] pushA,int [] popA) {
        Stack<Integer> s = new Stack<>();
        int i = 0, j = 0;
        while(i < pushA.length)
        {
            if (s.empty())
            {
                s.push(pushA[i++]);
            }
            if (s.peek() == popA[j])
            {
                s.pop();
                j++;
            }
            else
            {
                if (i < pushA.length)
                    s.push(pushA[i++]);
            }
        }
        while(!s.empty())
        {
            if (s.pop() != popA[j++])
                return false;
        }
        return true;
    }
}