题目
输入两个整数序列,第一个序列表示栈的压入顺序,请判断第二个序列是否可能为该栈的弹出顺序。假设压入栈的所有数字均不相等。例如序列1,2,3,4,5是某栈的压入顺序,序列4,5,3,2,1是该压栈序列对应的一个弹出序列,但4,3,5,1,2就不可能是该压栈序列的弹出序列。(注意:这两个序列的长度是相等的)
思路
- 把题目给的正例和反例的过程画出来看一下就懂了。
- 有两个数组,pushV表示入栈顺序,popV表示出栈顺序。
- 想出栈得先进站,假如popV的第一个元素为a,则需要pushV中的元素不停进栈直到遇到a元素。
- 然后popV的第一个元素为b,需要pushV中的元素不停进栈直到遇到b元素。
- 也有可能pushV的元素都进栈了,继续遍历popV,如果当前popV的元素等于栈顶元素,则栈pop。如果不相等说明不能这样pop,就返回false。
代码
class Solution {
public:
bool IsPopOrder(vector<int> pushV,vector<int> popV) {
stack<int> s;
bool res = false;
int pushIndex = 0;
int popIndex = 0;
while ( popIndex < popV.size() && pushIndex < pushV.size() ) {
// pushV中找到popv[i];
while ( pushIndex < pushV.size() ) {
s.push( pushV[pushIndex] );
++pushIndex;
if ( pushV[pushIndex-1] == popV[popIndex] ){
s.pop();
break;
}
}
++popIndex;
}
while ( popIndex < popV.size() && popV[popIndex] == s.top() ) {
s.pop();
++popIndex;
}
if ( s.empty() ) res = true;
return res;
}
};