题目描述
输入两个整数序列,第一个序列表示栈的压入顺序,请判断第二个序列是否为该栈的弹出顺序。假设压入栈的所有数字均不相等。例如序列1,2,3,4,5是某栈的压入顺序,序列4,5,3,2,1是该压栈序列对应的一个弹出序列,但4,3,5,1,2就不可能是该压栈序列的弹出序列。(注意:这两个序列的长度是相等的)
解题思路
- 需要借助一个辅助栈,i,j分别为pushV和popV当前值的下标,依次对需要压进栈的值和弹出值进行判断:
- 如果两者相等,说明需要将该值压入后再弹出,则i++,j++即可;
- 如果两者不相等,则判断popV[j]和当前栈中栈顶元素是否相等,相等的话将栈顶元素弹出;不相等的话,将pushV[i]压入栈;
- 如果pushV中的值已经遍历完成,则只需要判断popV[j]和当前栈中栈顶元素是否相等,相等的话将栈顶元素弹出;不相等的话,说明该序列不是对应的弹出序列,return false;
- 直到popV中的值也遍历完成,说明该序列是对应的弹出序列,return true。
代码
class Solution { public: bool IsPopOrder(vector<int> pushV,vector<int> popV) { if( pushV.size() != popV.size() ) return false; int i = 0; int j = 0; stack<int> s; while( 1 ) { if( j == popV.size() ) return true; if( i == pushV.size() ) { if( popV[j] == s.top() ) { s.pop(); j++; } else return false; } else if( pushV[i] == popV[j] ) { i++; j++; } else if( !s.empty() && popV[j] == s.top() ) { s.pop(); j++; } else { s.push(pushV[i]); i++; } } return false; } };
剑指offer思路
和上述思路类似,但是更有条理更容易理解一些, 同样需要建立一个辅助栈,具体步骤如下:
(1)如果下一个弹出的数字刚好是栈顶数字,那么直接弹出;
(2)如果一下个弹出的数字不在栈顶,则把压栈序列中还没有入栈的数字压入辅助栈,直到把下一个需要弹出的数字压入栈顶;
(3)如果所有数字都压入栈仍然没有找到下一个弹出的数字,那么该序列不可能是一个弹出序列。
代码
class Solution { public: bool IsPopOrder(vector<int> pushV,vector<int> popV) { stack<int> st; int id=0; for(int i=0;i<popV.size();++i){ while(st.empty()||st.top()!=popV[i]){ st.push(pushV[id++]); if(id>pushV.size()){ return false; } } st.pop(); } return st.empty(); };