牛客网《剑指Offer》 编程 21.栈的压入、弹出序列

题目描述

输入两个整数序列，第一个序列表示栈的压入顺序，请判断第二个序列是否可能为该栈的弹出顺序。假设压入栈的所有数字均不相等。例如序列1,2,3,4,5是某栈的压入顺序，序列4,5,3,2,1是该压栈序列对应的一个弹出序列，但4,3,5,1,2就不可能是该压栈序列的弹出序列。（注意：这两个序列的长度是相等的）。

解题思路

需要设置一个辅助栈。测试某个序列是否是一个出栈序列，可以通过测试原序列能否按照出栈序列的顺序出栈。

原序列和出栈序列分别存储在数组pushV和popV数组里。

在原序列下标小于原序列大小的时候：

    当栈为空时，原序列元素入栈。原序列元素下标加一。

    当栈顶元素与出栈序列下标最小的元素大小相等的时候，弹栈，同时待测序列下标加一。

    当栈顶元素与出栈序列的元素下标最小元素不相等的时候，压栈同时原序列元素下标加一。

此时所有元素都入栈过了，如果元素的出栈顺序确实是出栈序列接下来的序列，这说明原序列能够按照出栈序列出栈；

如果元素的出栈顺序与出栈序列接下来的序列不符合，说明原序列不能按照出栈序列出栈。

代码实现

class Solution {
public:
    bool IsPopOrder(vector<int> pushV,vector<int> popV) {
        int idx_push=0;
		int idx_pop = 0;
		stack<int> stk;
		while(idx_push<pushV.size())
		{
			if (stk.empty()) {
				stk.push(pushV[idx_push]);
				idx_push++;
				continue;
			}
			if (stk.top() == popV[idx_pop]) {
				stk.pop();
				idx_pop++;
			}
			else {
				stk.push(pushV[idx_push]);
				idx_push++;
			}
			//cout << idx_push << endl;
		}
		while (idx_pop<popV.size() && popV[idx_pop] == stk.top()) {
			stk.pop();
			idx_pop++;
		}
		if (idx_pop == popV.size()) { return true; }
		else { return false; }
    }
};