描述
用两个栈来实现一个队列,使用n个元素来完成 n 次在队列尾部插入整数(push)和n次在队列头部删除整数(pop)的功能。 队列中的元素为int类型。保证操作合法,即保证pop操作时队列内已有元素。
数据范围: n\le1000n≤1000
要求:存储n个元素的空间复杂度为 O(n)O(n) ,插入与删除的时间复杂度都是 O(1)O(1)
示例1
输入:
["PSH1","PSH2","POP","POP"]
返回值:
1,2
说明:
"PSH1":代表将1插入队列尾部 "PSH2":代表将2插入队列尾部 "POP“:代表删除一个元素,先进先出=>返回1 "POP“:代表删除一个元素,先进先出=>返回2
示例2
输入:
["PSH2","POP","PSH1","POP"]
返回值:
2,1
package com.example;
import java.util.*;
public class Main2 {
static Stack<Integer> stack1=new Stack<Integer>();
static Stack<Integer> stack2=new Stack<Integer>();
static String strInt="";
public static void main(String[] args){
Scanner sc=new Scanner(System.in);
while(sc.hasNext()){
strInt="";
String str=sc.nextLine();
int lenth=str.length();
String strString=str.substring(2,lenth-2);
String[] strList=strString.split("\",\"");
for(int i=0;i<strList.length;i++){
int lent=strList[i].length();
String sp=strList[i].substring(0,3);
if(sp.equals("POP")){
pop();
}else{
int si=Integer.parseInt(strList[i].substring(3,lent));
push(si);
}
}
int len=strInt.length();
System.out.println(strInt.substring(0,len-1));
}
}
//模拟队列pop
private static void pop() {
if (!stack2.empty()){
Integer s =stack2.pop();
strInt=strInt+s+",";
}
else{
while(!stack1.empty()){
Integer s=stack1.pop();
stack2.push(s);
}
Integer s =stack2.pop();
strInt=strInt+s+",";
}
}
//模拟队列pish
private static void push(int si) {
stack1.push(si);
}
}