剑指offer之面试题7:用两个栈实现队列

来源:互联网 发布:linux运维面试题技巧 编辑:程序博客网 时间:2024/05/03 16:59

题目描述

用两个栈来实现一个队列,完成队列的Push和Pop操作。 队列中的元素为int类型。

思路:push操作无所谓栈或队列,都是往里面加入元素。而区别在于pop操作,队列的pop操作取的是先push的元素,而栈pop的则是最后push的元素,怎样通过两个栈来实现队列的pop操作,是主要问题所在。举个例子,向stack1中逐个push进a,b,c,则stack1中元素{a,b,c},c位于栈顶,再将stack1中元素pop,并push进stack2中,则stack2元素{c,b,a},a位于栈顶,pop出的顺序为a,b,c,符合先进先出的队列。简而言之,stack2中的元素是符合队列pop顺序的,可以直接pop,而stack1中的元素要想按照队列的pop需先push到stack2中,即当stack2不为空时,可以直接pop,当stack2为空时,将stack1中的元素pop并push进stack2中,再pop。

说的有点乱,贴上代码,就会更容易理解:

import java.util.Stack;public class Solution {    public static Stack<Integer> stack1 = new Stack<Integer>();    public static Stack<Integer> stack2 = new Stack<Integer>();    public static void push(int node) {        stack1.push(node);    }    public static int pop() {        if(stack2.empty()){            while(!stack1.empty()){                stack2.push(stack1.pop());            }        }        if(stack2.empty()){            System.out.println("queue is empty");        }        return stack2.pop();    }    public static void main(String[] args){        push(1);        push(2);        push(3);        push(4);        System.out.println(pop());    }}

扩展:用两个队列实现一个栈

思路:栈的push操作,当两个队列都为空时,不妨将元素push进queue1;或者push进非空队列。栈的pop操作,根据队列的先进先出原则,先插入的元素在队首,后插入的在队尾,而栈的操作要求后进先出,即队尾的元素要先出,可以将非空队列的元素逐个出队并入队到空队列中,直到空队列剩余一个元素,此时删除这个元素即可。

贴出代码

package com.su.biancheng;import java.util.LinkedList;import java.util.Queue;/** * @title StackWithTwoQueues.java * @author Shuai * @date 2016-4-15下午9:00:25 */public class StackWithTwoQueues {    public static Queue<Integer> queue1=new LinkedList<Integer>();    public static Queue<Integer> queue2=new LinkedList<Integer>();    public static void push(int e){        //插入非空队列        if(queue2.isEmpty())            queue1.offer(e);        if(queue1.isEmpty())            queue2.offer(e);    }    public static int pop(){        //将非空队列的元素逐个出队并插入到空队列        //poll,获取并移除此队列的头,如果此队列为空,则返回 null。        //remove,获取并移除此队列的头。此方法与 poll 唯一的不同在于:此队列为空时将抛出一个异常。        if(!queue1.isEmpty()){            while(queue1.size()>1){                queue2.offer(queue1.poll());            }            return queue1.poll();        }        else if(!queue2.isEmpty()){            while(queue2.size()>1){                queue1.offer(queue2.poll());            }            return queue2.poll();        }        return 0;    }    public static void main(String[] args){        push(1);        push(2);        push(3);        push(4);        push(5);        System.out.println(pop());    }}
0 0
原创粉丝点击