试题来源:《程序员面试金典》
https://leetcode-cn.com/problems/sort-of-stacks-lcci/
栈排序: 编写程序,对栈进行排序使最小元素位于栈顶。最多只能使用一个其他的临时栈存放数据,但不得将元素复制到别的数据结构(如数组)中。该栈支持如下操作:push、pop、peek 和 isEmpty。当栈为空时,peek 返回 -1。

代码:
Java:

class SortedStack {

    Stack<Integer> p=new Stack<>();
    Stack<Integer> q=new Stack<>();

    public SortedStack() {

    }
    
    public void push(int val) {
        if(p.empty())  p.push(val);
        else{
            while(!p.empty()&&p.lastElement()<=val){
                q.push(p.lastElement());
                p.pop();
            }
            p.push(val);
            while(!q.empty()){
                p.push(q.lastElement());
                q.pop();
            }
        }
    }
    
    public void pop() {
        if(!p.empty())  p.pop();
    }
    
    public int peek() {
        return p.empty()?-1:p.lastElement();
    }
    
    public boolean isEmpty() {
        return p.empty();
    }
}

 

C++:

class SortedStack {
    stack<int> p;
public:
    SortedStack() {

    }
    
    void push(int val) {
        if(p.size()==0)  p.push(val);
        else{
            stack<int> q;
            while(!p.empty()&&p.top()<val){
                q.push(p.top());
                p.pop();
            }
            p.push(val);
            while(!q.empty()){
                p.push(q.top());
                q.pop();
            }
        }
    }
    
    void pop() {
        if(p.size()>0)  p.pop();
    }
    
    int peek() {
        return p.size()==0?-1:p.top();
    }
    
    bool isEmpty() {
        return p.empty();
    }
};

 

posted on 2020-02-27 14:20  nibolyoung  阅读(728)  评论(0编辑  收藏  举报