题目:
输入两个整数序列,第一个序列表示栈的压入顺序,请判断第二个序列是否为该栈的弹出顺序。假设压入栈的所有数字均不相等。
例如序列1,2,3,4,5是某栈的压入顺序,序列4,5,3,2,1是该压栈序列对应的一个弹出序列,但4,3,5,1,2就不可能是该压栈序列的弹出序列。
思路:
直观的想法就是模拟第一个序列的数字依次入栈,并按照第二个序列依次弹出,如果第一个序列全部入栈,并最后栈中元素都被弹出来,就说明第二个序列是该栈的弹出顺序。
- 建立一个辅助栈;
- 将第一个序列的数字压入辅助栈;
- 如果第二个序列的下一个弹出数字刚好是栈顶数字,则直接弹出,第二个序列弹出当前数字,辅助栈也弹出该数字;
- 否则,就把第一个序列中尚未入栈的数字压入辅助栈,直到把第二个序列中下一个需要弹出的数字压入栈顶为止。
- 如果第一个序列的所有数字都已经入栈了,仍然没有找到下一个弹出的数字,那么第二个序列不可能是一个弹出序列。
代码:
#include <iostream> #include <stack> using namespace std; bool IsPopOrder(const int* push,const int* pop,int length){ bool bPossible=false; if(push!=NULL && pop!=NULL && length>0){ std::stack<int> stackData; int i=0; int j=0; while(j<length){ while(stackData.empty() || stackData.top()!=pop[j]){ if(i>length-1) break; stackData.push(push[i]); ++i; } if(stackData.top()!=pop[j]) break; stackData.pop(); ++j; } if(stackData.empty() && j==length) bPossible=true; } return bPossible; } int main() { int A[]={1,2,3,4,5}; int B[]={4,5,3,2,1}; int length=sizeof(A)/sizeof(A[0]); cout << IsPopOrder(A,B,length) << endl; return 0; }
在线测试OJ:
http://www.nowcoder.com/books/coding-interviews/d77d11405cc7470d82554cb392585106?rp=1
AC代码:
class Solution { public: bool IsPopOrder(vector<int> pushV,vector<int> popV) { int length=pushV.size(); if(pushV.size()>0 && popV.size()>0 && pushV.size()==popV.size()){ std::stack<int> stackData; int i=0; int j=0; while(j<length){ while(stackData.empty() || stackData.top()!=popV[j]){ if(i>length-1) break; stackData.push(pushV[i]); ++i; } if(stackData.top()!=popV[j]) break; stackData.pop(); ++j; } if(stackData.empty() && j==length) return true; } return false; } };