码迷,mamicode.com
首页 > 其他好文 > 详细

LintCode Implement Queue by Two Stacks

时间:2016-08-21 15:10:03      阅读:179      评论:0      收藏:0      [点我收藏+]

标签:

1. stack(先进后出):

pop 拿出并返回最后值; peek 返回最后值; push 加入新值在后面并返回此值。

2. queue(先进先出) :

poll = remove 拿出并返第一个值; element = peek 返第一个值; add = offer 加入新值在后面并返回true/false。

做此题时, 第一个stack为基础, 第二个stack为媒介来颠倒顺序, 加时从第一个加, 取时从第二个取。

public class Queue {
    private Stack<Integer> stack1;
    private Stack<Integer> stack2;

    public Queue() {
       stack1 = new Stack<Integer>();
       stack2 = new Stack<Integer>();
       // do initialization if necessary
    }
    
    public void push(int element) {
        stack1.push(element);
        // write your code here
    }

    public int pop() {
        if(!stack2.empty()){
            return stack2.pop();
        } else{
            while(stack1.size() > 0){
                stack2.push(stack1.pop());
            }
            if(!stack2.empty()){
                return stack2.pop();
            } else return -1;
        }
        // write your code here
    }

    public int top() {
        if(!stack2.empty()){
            return stack2.peek();
        } else{
            while(stack1.size() > 0){
                stack2.push(stack1.pop());
            }
            if(!stack2.empty()){
                return stack2.peek();
            } else return -1;
        }
        // write your code here
    }
}

 

LintCode Implement Queue by Two Stacks

标签:

原文地址:http://www.cnblogs.com/LittleAlex/p/5792646.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!