题目描述:
输入两个整数序列,第一个序列表示栈的压入顺序,请判断第二个序列是否为该栈的弹出顺序。假设压入栈的所有数字均不相等。例如序列1,2,3,4,5是某栈的压入顺序,序列4,5,3,2,1是该压栈序列对应的一个弹出序列,但4,3,5,1,2就不可能是该压栈序列的弹出序列
这道题考察的是STL中的栈,与ZOJ上的火车进站问题一样。
class Solution {
public:
bool IsPopOrder(vector<int> pushV,vector<int> popV) {
int len1 = pushV.size(), len2 = popV.size();
if(len1 == 0 || len2 == 0)
return false;
if(len1 == 1 && len2 == 1)
{
if(pushV[0] == popV[0])
return true;
else
return false;
}
if(len1 != len2)
return false;
stack<int> s;
int i = 0, j = 0;
while(i < len1 && j < len1)
{
if(s.empty() || pushV[i] != popV[j])
{
s.push(pushV[i]);
i++;
}
if(pushV[i] == popV[j])
{
i++;
j++;
while(!s.empty() && popV[j] == s.top() && j < len1)
{
s.pop();
j++;
}
}
}
if(j == len1)
return true;
else
return false;
}
};