剑指offer-合法的出栈序列

问题

题目：[合法的出栈序列]

思路

这个题我模拟了，写的有点麻烦。
基本思路是考虑出栈序列，对于每一个出栈元素，进行模拟，因为如果当前元素要出栈，那么证明他以及他之前的元素都在栈内。所以，将之前的元素入栈即可。
可以写如下判断逻辑：
1.如果当前出栈元素在栈里面（其实用集合判断），只需判断栈顶和当前出栈元素是否一样。如果不一样，返回false.
2.如果当前出栈元素不在栈里面，只能是从入栈的序列里面进行扫描，直到遇到当前元素，一直入栈。

这个有个异常数据就是如果入栈序列和出在序列里面存在不同的元素，这个是我之前没有考虑到的。比如[1,2],[2,3]

代码

class Solution {
public:
    bool IsPopOrder(vector<int> pushV,vector<int> popV) {
        set<int> visit;
        stack<int> stk;

        int sz_push = pushV.size();
        if(!sz_push) return true;
        int sz_pop = popV.size();

        int j = 0;
        for(int i = 0; i < sz_pop; ++i){
            if( visit.find(popV[i]) == visit.end() ){
                while(j < sz_push ){
                    stk.push( pushV[j] );   
                    visit.insert(pushV[j]);
                    if( pushV[j] == popV[i] )
                        break;
                    ++j;
                }
                if( j == sz_push )
                    return false;
                stk.pop();
                ++j;
            }
            else{
                if( popV[i] == stk.top() )
                    stk.pop();
                else
                    return false;
            }
        }
        return true;
    }
};