【剑指Offer-举例让抽象问题具体化】面试题31:栈的压入、弹出序列
题目描述
输入两个整数序列,第一个序列表示栈的压入顺序,请判断第二个序列是否可能为该栈的弹出顺序。假设压入栈的所有数字均不相等。例如序列1,2,3,4,5是某栈的压入顺序,序列4,5,3,2,1是该压栈序列对应的一个弹出序列,但4,3,5,1,2就不可能是该压栈序列的弹出序列。(注意:这两个序列的长度是相等的)
思路
思路如下:构建一个辅助栈s,如果s的栈顶元素等于下一个弹出的数字,则将s的栈顶元素弹出;如果s为空或者s的栈顶元素不等于下一个弹出的数字,则将入栈序列中的元素依次压入栈中直到当前栈顶元素和下一个弹出的元素相等。如果将入栈序列全部压入栈中都没有找到和下一个弹出元素相等的元素,则该弹出序列不可能。
class Solution {
public:
stack<int> s; //辅助栈
bool IsPopOrder(vector<int> pushV,vector<int> popV) {
if(pushV.empty() || popV.empty())
return false;
int lenPush = pushV.size();
int lenPop = popV.size();
int curPush = 0;
int curPop = 0;
while(curPop<lenPop){
while(s.empty() || s.top()!=popV[curPop]){
if(curPush==lenPush) //入栈序列全部压入栈中也没有遇到和下一个弹出元素相等的元素
return false;
s.push(pushV[curPush]);
curPush++;
}
if(s.top()!=popV[curPop]) //栈顶元素不等于下一个要弹出的元素
return false;
s.pop();
curPop++;
}
if(s.empty() && curPop==lenPop) //栈为空并且出栈序列被遍历了一遍
return true;
return false;
}
};