1. 程式人生 > >判斷給定序列是否是對應入棧序列的出棧序列(C++)

判斷給定序列是否是對應入棧序列的出棧序列(C++)

題目:

輸入兩個整數序列,第一個序列表示棧的壓入順序,請判斷第二個序列是否為該棧的彈出順序。

(假設壓入棧的所有數字均不相等)

如 序列1,2,3,4,5是某棧的壓入順序,序列4,5,3,2,1是該壓棧序列對應的一個彈出序列,但4,3,5,1,2不可能是該壓棧序列的彈出序列。

思路:

使用一個輔助棧,依次將壓棧序列的元素入棧,同時判斷出棧序列的元素與棧頂元素是否相等。

全部壓棧結束後,如果輔助棧內還有剩餘元素,說明給定的出棧序列不能對應壓棧序列。

貼程式碼:

#include <iostream>

#include <vector>
#include <stack>

using namespace std;

class StackPushPop {
public:
	bool IsPopOrder(vector<int> pushV,vector<int> popV) 
	{
		if (pushV.size() == 0 || pushV.size() != popV.size())
		{
			return false;
		}

		stack<int> stk;
		int j = 0;

		for (int i = 0; i < pushV.size(); ++i)
		{
			stk.push(pushV[i]);

			while(!stk.empty() && stk.top() == popV[j])
			{
				stk.pop();
				j++;
			}
		}

		return stk.empty();
	}
};

int main(void)
{
	int pushV[] = {1,2,3,4,5};
	//int popV[] = {4,5,3,2,1};
	int popV[] = {4,5,3,1,2};

	vector<int> v1(pushV, pushV + 5);
	vector<int> v2(popV, popV + 5);

	StackPushPop spp;

	bool res = spp.IsPopOrder(v1, v2);

	if (res)
	{
		cout<<"true"<<endl;
	}
	else
	{
		cout<<"false"<<endl;
	}

	return 0;
}