剑指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;
}