题目:
输入两个整数序列,第一个序列表示栈的压入顺序,请判断第二个序列是否为该栈的弹出顺序。假设压入栈的所有数字均不相等。例如序列1、2、3、4、5是某栈的压栈序列,序列4、5、3、2、1是该压栈序列对应的一个弹出序列,但4、3、5、1、2就不可能是该压栈序列的弹出序列。
分析两个例子,找到规律:
1、弹出序列4、5、3、2、1
第一个希望被弹出的数字是4,因此4需要先压入辅助栈中,而在4之前1、2、3需要先压入栈中。此时栈里含4个数字,其中4位于栈顶。把4弹出栈,剩下1、2、3,但接下来需要弹出5,由于栈顶元素不为5,因此在还未入栈的数字中找到5压栈。此时5位于栈顶,可以弹出。接下来依次弹出3、2、1,由于每次操作前它们都位于栈顶,因此直接弹出即可。
2、弹出序列4、3、5、1、2
第一个希望弹出的数字是4,与前面相同。弹出4后,希望弹出3,3位于栈顶,可以直接弹出。接下来希望弹出5,在还未入栈的数字中找到5压栈,此时5位于栈顶,可以弹出。接下来希望弹出1,但是位于栈顶的是2,所以需要在还未入栈的元素中找1,发现没有,因此该序列不是弹出序列。
总结:如果下一个弹出的数字刚好是栈顶数字,那么直接弹出。如果下一个弹出的数字不在栈顶,就在还未入栈的数字里查找,如果找不到,则证明该序列不是一个弹出序列。
bool IsPopOrder(const int* pPush,const int* pPop,int nLength)
{
bool bPossible=false;
if(pPush!=NULL && pPop!=NULL && nLength>0)
{
//将指针指向压入序列
const int* pNextPush=pPush;
//将指针指向弹出序列
const int* pNextPop=pPop;
std::stack<int>stackData;
//还没完全弹出时
while(pNextPop-pPop<nLength)
{
//当栈空或者栈顶元素不是下一个期待弹出的元素时
while(stackData.empty() || stackData.top()!=*pNextPop)
{
//当已经全部压入
if(pNextPush-pPush==nLength)
break;
//将压入序列中的下一个元素压栈
stackData.push(*pNextPush);
pNextPush++;
}
if(stackData.top()!=*pNextPop)
break;
stackData.pop();
pNextPop++;
}
//当栈空且全部弹出,说明序列正确
if(stackData.empty() && pNextPop-pPop==nLength)
bPossible=true;
}
return bPossible;
}