每天一道算法题之两个栈实现一个队列

题目描述:

用两个栈实现一个队列。队列的声明如下，请实现它的两个函数 appendTail 和deleteHead ，分别完成在队列尾部插入整数和在队列头部删除整数的功能。(若队列中没有元素，deleteHead 操作返回 -1 )¹

• 解析: 一个栈(in)用来入队列,一个栈(out)用来出队列
  • 入队列:直接入in栈
  • 出队列:
    首先判断out栈是否为空
    若不为空,直接出栈,
    若out栈为空,则判断in栈是否为空
    若in栈为空,则队列为空,返回-1
    若in栈不为空,则将in栈中元素出栈,并压到out栈,随后输出out栈顶元素
• 代码部分:

import java.util.*;
public class Cqueue {

	Stack<Integer> in;
	Stack<Integer> out;
	public Cqueue {
    	in = new Stack<>();
    	out = new Stack<>();
	}
	//入栈
	public void appendTail(int value) {
    	in.push(value);
	}
	//出栈
	public int deleteHead() {
	
    	if(!out.isEmpty()){//out栈非空直接出栈
    	
        	return out.pop();
        
    	}else{//out为空
    	
        	if(!in.isEmpty()){//in栈非空,说明队列里还有元素
            	while(!in.isEmpty()){
                	out.push(in.pop());
            	}
            	return out.pop();
        	}else{//in栈为空,说明队列里没有元素
            	return -1;
        	}
        
    	}
	}
}

---

1. https://leetcode-cn.com/problems/yong-liang-ge-zhan-shi-xian-dui-lie-lcof/ ↩︎