两个栈实现队列(腾讯笔试题)



第四题:两个栈实现队列

用两个栈实现队列,支持队列的基本操作。
输入描述:
第一行输入一个整数N,表示对队列进行的操作总数。下面N行每行输入一个字符串S,表示操作的种类。
如果s为"add",则后面还有一个整数x表示向队列尾部加入整数X。
如果s为"poll",则表示弹出队列头部操作。
如果s为"peek",则表示询问当前队列中头部元素是多少。
1 < N< 1000000, -1000000 < X < 1000000
数据保证没有不合法的操作。
输出描述:
对于每一个为"peek"的操作,输出一行表示当前队列中头部元素是多少。

package offer;

import java.util.LinkedList;
import java.util.*;
public class Queue {
public static void main(String[] args) {
Scanner scan = new Scanner(System.in);
int N = scan.nextInt();
LinkedList<String> queue = new LinkedList<>();
for (int i = 0; i < N; i++) {
String str = scan.next();
switch (str) {
case "add":
String x = scan.next();
queue.add(" " + x);
break;
case "poll":
queue.pop();
break;
case "peek":
System.out.println(queue.peek());
break;
default:
break;
}
}
}
}

用例输入输出:
6
add 1
add 2
add 3
peek
 1
poll
peek
 2
 

猜你喜欢

转载自www.cnblogs.com/zwjnb/p/12784313.html