题目:
给出一个栈的压入序列,判断给出的弹出序列是否正确。
代码如下:
#include <iostream>
#include <stdio.h>
#include <stack>
using namespace std;
bool IsPopOrder(const int* push_order,const int* pop_order,int length)
{
if( push_order == NULL || pop_order == NULL || length < 1)
return false;
stack <int> data;
int i = 0 , j =0;
while( i < length)
{
data.push(push_order[i]);
if(pop_order[j] == data.top())
{
data.pop();
j++;
}
i++;
}
if(data.size() == 0)
return true;
while(!data.empty())
{
if(data.top() == pop_order[j])
{
data.pop();
j++;
}
else
return false;
}
return true;
}
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);
}
void Test2()
{
const int nLength = 5;
int push[nLength] = {1, 2, 3, 4, 5};
int pop[nLength] = {3, 5, 4, 2, 1};
Test("Test2", push, pop, nLength, true);
}
void Test3()
{
const int nLength = 5;
int push[nLength] = {1, 2, 3, 4, 5};
int pop[nLength] = {4, 3, 5, 1, 2};
Test("Test3", push, pop, nLength, false);
}
void Test4()
{
const int nLength = 5;
int push[nLength] = {1, 2, 3, 4, 5};
int pop[nLength] = {3, 5, 4, 1, 2};
Test("Test4", push, pop, nLength, false);
}
// push和pop序列只有一个数字
void Test5()
{
const int nLength = 1;
int push[nLength] = {1};
int pop[nLength] = {2};
Test("Test5", push, pop, nLength, false);
}
void Test6()
{
const int nLength = 1;
int push[nLength] = {1};
int pop[nLength] = {1};
Test("Test6", push, pop, nLength, true);
}
void Test7()
{
Test("Test7", NULL, NULL, 0, false);
}
int main()
{
Test1();
Test2();
Test3();
Test4();
Test5();
Test6();
Test7();
system("PAUSE");
return 0;
}