剑指offer之栈的压入、弹出序列

题目描述

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

解题思路

模拟进栈和出栈即可

Code

• cpp

class Solution {
public:
    bool IsPopOrder(vector<int> pushV,vector<int> popV) {
        if(!pushV.size()) return false;
        stack<int> s;
        int i = 0, j = 0;
        for(; j<popV.size(); ) {
            if(!s.empty() && s.top() == popV[j]){
                s.pop(); 
                j++;
            } else if(i<pushV.size() && pushV[i] == popV[j]) {
                i++, j++;
            } else {
                if(i < pushV.size()) {
                    s.push(pushV[i++]);
                } else {
                    break;
                }
            }
        }
        return s.empty();
    }
};

• java

import java.util.ArrayList;
import java.util.LinkedList;

public class Solution {
    public boolean IsPopOrder(int [] pushA,int [] popA) {
        int len1 = pushA.length, len2 = popA.length;
        LinkedList<Integer> stack = new LinkedList<Integer>();
        for(int i = 0, j = 0; i<len1; i++) {
            if(pushA[i] == popA[j]) {
                j++;
            } else {
                stack.add(pushA[i]);
            }
            while(j < len2 && !stack.isEmpty() && stack.getLast() == popA[j]) {
                j++;
                stack.removeLast();
            }
        }
        return stack.isEmpty();
    }
}

总结

• LinkedList作为双端队列也可以用作栈，不过push和pop操作都是在队首上进行的，因此操作的时候还是指定位置比较好，比如addLast和removeLast