栈的压入、弹出序列 【c++】
本博文是关于牛客网《剑指offer》里相关题目的刷题笔记,大家可以戳链接:
剑指offer_编程题_牛客网
栈的压入、弹出序列
题目描述
输入两个整数序列,第一个序列表示栈的压入顺序,请判断第二个序列是否可能为该栈的弹出顺序。假设压入栈的所有数字均不相等。例如序列1,2,3,4,5是某栈的压入顺序,序列4,5,3,2,1是该压栈序列对应的一个弹出序列,但4,3,5,1,2就不可能是该压栈序列的弹出序列。(注意:这两个序列的长度是相等的)
思路
判断第二个序列是否可能为该栈的弹出顺序,过程如下:
-
当1,2,3,4,5是某栈的压入顺序时,判断4,5,3,2,1 是否可能是其相应的弹出序列?
第一个希望弹出4,则4应该在栈顶,而压入4之前,1,2,3已经入栈,4出栈后,希望弹出5, 5不在栈顶,继续压入,直至栈顶为5,弹出;希望弹出3,发现3在栈顶,弹出3;希望弹出2,1,发现都在栈顶。则该序列是一个弹出序列。
-
当1,2,3,4,5是某栈的压入顺序时,判断4,3,5,1,2 是否可能是其相应的弹出序列?
第一个希望弹出4,则4应该在栈顶,而压入4之前,1,2,3已经入栈。4出栈后,希望弹出3,发现3在栈顶,弹出 3;希望弹出5, 5不在栈顶,继续压入直至栈顶为5,弹出 5;希望弹出1,发现1不在栈顶,继续压入输入,发现已无输入。则该序列不是一个弹出序列。
总结:借助辅助栈tempt,如果下一个希望弹出的数字刚好是栈顶数字,则直接弹出;如果下一个数字不在栈顶,我们把压栈序列中还没有入栈的数字压入栈,直至把下一个需要弹出的数字压入栈顶为止。如果所有数字已入栈,仍没有找到下一个要弹出的数字,那么该序列不是一个弹出序列。
class Solution {
public:
bool IsPopOrder(vector<int> pushV,vector<int> popV) {
if(pushV.size() == 0 || popV.size() == 0) return false;
stack<int> tempt;//辅助栈
int j = 0;
//将pushV中的序列压入栈
for(int i = 0;i < pushV.size();i++)
{
tempt.push(pushV[i]);
while(!tempt.empty() && tempt.top() == popV[j])
{
tempt.pop();
j++;
}
}
return j == popV.size();
//return st.empty(); 也可
}
};