面试题22:栈的压人、弹出序列
题目:输入两个整数序列,第一个序列表示栈的压人顺序,请判断第一个序列是否为该栈的弹出顺序。假设压人栈的所有数字均不相等。例如序列1、2、3、4、5是某栈的压栈序列,序列4、5、3、2、1是该栈序列对应的一个弹出序列,但是4、3、5、1、2就不可能是该压栈序列的弹出序列。
预备知识:
STL中我们实现了一个很好的栈stack,这需要我们引入头文件<stack>
template <class T, class Container = deque<T> > class stack;容器类的必须操作:
- back
- push_back
- pop_back
思路:解决这类问题需要建立一个辅助栈,把输入的第一个序列中的数字依次压入到辅助栈,并按照第二个序列的顺序依次从该栈中弹出。具体可以通过举个例子寻找规律。如果下一个弹出的数字刚好是栈顶数字,那么直接弹出;如果下一个弹出的数字不在栈顶,那么我们把压栈序列中还没有入栈的数字压入辅助栈,直到把下一个需要弹出的数字压入栈顶为止;如果所有的数字都压入栈了,仍然没有找到下一个弹出的数字,那么该序列不可能是一个弹出序列!!
算法实现:
// 面试题22.cpp : 定义控制台应用程序的入口点。
//
#include "stdafx.h"
#include <stack> //STL: stack 栈
//栈的压入、弹出序列(两个整数序列)
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 ++;
} //while
if(stackData.top() != *pNextPop)
break;
//如果下一个弹出的数字刚好是栈顶数字,那么直接弹出
stackData.pop(); //出栈
pNextPop ++;
} //while
if(stackData.empty() && pNextPop - pPop == nLength)
bPossible = true;
} //if
return bPossible;
}
//*************测试代码***************
void Test(char* testName, const int* pPush, const int* pPop, int nLength, bool expected)
{
if(testName != NULL)
printf("%s begins:", testName);
if(IsPopOrder(pPush, pPop, nLength) == expected)
printf("Passed.\n");
else
printf("failed.\n");
}
void Test1()
{
const int nLength = 5;
int push[nLength] = {1,2,3,4,5};
int pop[nLength] = {4,5,3,2,1};
Test("Test1", push, pop, nLength, true);
}
int _tmain(int argc, _TCHAR* argv[])
{
Test1();
return 0;
}
注意事项:
测试用例:输入两个数组含有多个数字或者只有一个数字;第二个数组是或者不是第一个数组表示的压入序列对应的栈的弹出序列;输入特殊序列,如输入两个NULL指针。