题目:输入两个整数序列,第一个序列表示栈的压入顺序,请判断第二个序列是否为该栈的弹出顺序。假设压入栈的所有数字均不相等。例如序列1、2、3、4、5是某栈的压栈序列,序列4、5、3、2、1就是该栈对应的一个弹出序列,但4、3、5、1、2就不可能是该压栈序列的弹出序列。
思路:
使用辅助站,根据模拟弹出序列和压入序列
编译环境:ArchLinux+Clang3.3,C++11
实现一:
#include <iostream>
#include <stack>
#include <cassert>
using namespace std;
bool isPopOrder(const int *pPush, const int *pPop, int len)
{
bool ret = false;
if (pPush && pPop && len > 0) {
const int *pNextPush = pPush;
const int *pNextPop = pPop;
stack<int> st;
while (pNextPop < pPop + len) {
while ( (pNextPush < pPush + len)
&& (st.empty() || st.top() != *pNextPop) ) { // 若栈顶不等于当前元素,则持续入栈直到相等或者全部入栈
st.push(*pNextPush++);
}
if (st.top() != *pNextPop) // 实际上是判断上面循环退出条件之一
break;
st.pop();
pNextPop++;
}
if (st.empty() && pNextPop == pPop+len) // 成功条件
ret = true;
}
return ret;
}
int main()
{
int pPush[]{1,2,3,4,5};
int pPop[]{4,5,3,2,1};
int pErr[]{4,3,5,1,2};
assert(isPopOrder(pPush, pPop, sizeof(pPush)/sizeof(int)));
assert(!isPopOrder(pPush, pErr, sizeof(pErr)/sizeof(int)));
}