剑指offer-刷题笔记-中难题-JZ31 栈的压入、弹出序列

剑指offer-刷题笔记-中难题–JZ31 栈的压入、弹出序列

版本1-自己写的，可以通过部分例题，但是由于只能从第一个元素考虑，不完善

class Solution {
public:
    bool IsPopOrder(vector<int> pushV,vector<int> popV) {
       
        if(pushV.size() != 0 && popV.size() == 0)
        {
            return false;    
        }
        if(pushV.size() == 0 && popV.size() != 0)
        {
            return false; 
        }
        
        if(pushV.size() == 1 && popV.size() == 1)
        {
            if(pushV[0] == popV[0])
            {
                return true; 
            }else{
                return false; 
            }
            
        }
        //pop一个值x，如果后面的值在push顺序中比X靠后，后面pop位置随意
        //如果后面的值在push顺序中比X靠前，必须按照push顺序相反
        //辅助栈
        stack<int> s;
        //int count;
        for(int i = 0;i<pushV.size();i++)
        {
            //pop第一个值前的数字都在栈中
            if(pushV[i] != popV[0])
            {
                s.push(pushV[i]);
            }
            else if(pushV[i] == popV[0])
            {
                 //count = i;
                 s.push(pushV[i]);
                 break;
            }
        }
        
        //cout<<s.size()<<endl;
        
        for(int j = 0;j < popV.size();j++)
        {
            if(!s.empty())
            {
                if(popV[j] == s.top())
                {
                    s.pop();
                }
            }

        }
        //cout<<s.size()<<endl;
        
        if(s.size() == 0)
        {
            return true;
        }else{
            return false;
        }
        
    }
};

版本2

class Solution {
public:
    bool IsPopOrder(vector<int> pushV,vector<int> popV) {
        int n = pushV.size();
        //辅助栈
        stack<int> s;
        //遍历入栈的下标
        int j = 0;
        //遍历出栈的数组
        for(int i = 0; i < n; i++){
            //入栈：栈为空或者栈顶不等于出栈数组
            while(j < n && (s.empty() || s.top() != popV[i])){
                s.push(pushV[j]);
                j++;
            }
            //栈顶等于出栈数组
            if(s.top() == popV[i])
                s.pop();
            //不匹配序列
            else
                return false;
        }
        return true;
    }
};