剑指Offer_面试题22_栈的压入弹出序列

题目描述

输入两个整数序列，第一个序列表示栈的压入顺序，请判断第二个序列是否为该栈的弹出顺序。假设压入栈的所有数字均不相等。例如序列1,2,3,4,5是某栈的压入顺序，序列4，5,3,2,1是该压栈序列对应的一个弹出序列，但4,3,5,1,2就不可能是该压栈序列的弹出序列。（注意：这两个序列的长度是相等的）
分析：以前在做数据结构的题目时，判断是不是出栈序列是手动模拟入栈出栈的。那么其实写程序判断的时候也一样，只是需要仔细分析一下条件是什么。
就拿1,2,3,4,5 序列 4,3,5,1,2来说，第一个输出元素是4，栈顶元素不是4或者栈为空，需要在输入序列中找到4，每遍历一个入栈一个，直至找到4并入栈。最后满足条件应该是栈为空且输出序列遍历结束。
代码：
#include <vector>
#include <stdio.h>
#include <stack>
#include <iostream>
using namespace std;

bool IsPopOrder(vector<int> pushV, vector<int> popV) {
if (pushV.empty() || popV.empty() || pushV.size() != popV.size())
return false;
int len = pushV.size();
int i = 0;
int j = 0;
bool result = false;
stack<int> data;

while (j < len)
{
//栈空或者输出元素不在栈顶
while (data.empty() || data.top() != popV[j])
{
if (i == len) break; //元素全部压完
data.push(pushV[i++]); //入栈
}

//经过while循环还没找到输出元素
if (data.top() != popV[j]) break;

//找到的话
++j;
data.pop();
}

if (data.empty() && j == len)
result = true;

return result;
}

int main()
{
vector<int> pushV = {1,2,3,4,5};
vector<int> popV = {4,5,3,2,1};
cout << IsPopOrder(pushV, popV) << endl;

getchar();
return 0;
}