我有一个模拟队列的程序,其中有两种类型的输入 join
和 leave
。
join
输入附加一个整数,表示一个人的高度,如 join 180
,而 leave
输入附加一个整数, 表示离开队列的人数 leave 3
.
每次调用加入或离开查询时,程序应返回可以查看队列前面的人数。如果一个人在他/她前面的队列中的每个人都比他/她矮,或者如果他/她是队列中的第一个人,则他/她可以看到前面。
输入/输出的示例如图所示。
这是我的代码,我创建了两个堆栈来存储队列中的人数,以及可以查看前面的人数:
private void run() {
Scanner sc = new Scanner(System.in);
int no_queries = Integer.parseInt(sc.nextLine());
for(int i = 0; i < no_queries; i++){
String[] queries = sc.nextLine().split(" ");
int value = Integer.parseInt(queries[1]);
switch(queries[0]){
case "join":
if(!height.empty()){
if(height.peek() < value){
height.add(value);
}
}else{
height.add(value);
}
stack.add(value);
System.out.println(height.size());
break;
case "leave":
for(int j = 0; j < value; j++){
stack.pop();
}
//how do i remove the height stack?
System.out.println(height.size());
break;
}
}
}
这是我的输出,最后三个条目是错误的:
1
0
1
1
1
2
3
2
2
0
1
0
我的问题在于高度堆栈。当给出 leave 3
查询时,整数 160
、170
和 165
将从队列堆栈中删除。
但是,高度堆栈只包含两个整数 160
和 170
并且这两个值将被查询删除,尽管高度为 的人160
不应该从队列中移除,因为他在前面而不是离开的 3 个中的一部分。
我考虑过使用 Person 对象来存储最高的人变量,但是有没有更优雅的方法来解决这个问题?
最佳答案
我认为您可以使用 Person 对象,而不是存储最高的人,但可以比较两个堆栈中的值。我的意思是,当您执行 stack.pop();
时,您需要一种方法来检查您是否从堆栈中删除了最高人物对象,如果是,则也将其从最高人物堆栈中删除。
static class Person {
int height;
Person (int height) {
this.height = height;
}
}
private static void run() {
Stack<Person> stack = new Stack<>();
Stack<Person> highest = new Stack<>();
Scanner sc = new Scanner(System.in);
int n = Integer.parseInt(sc.nextLine());
for (int i = 0; i < n; i++) {
String[] queries = sc.nextLine().split(" ");
int value = Integer.parseInt(queries[1]);
switch (queries[0]) {
case "join":
Person person = new Person(value);
if (highest.empty()) {
highest.push(person);
} else if (value > highest.peek().height) {
highest.push(person);
}
stack.push(person);
System.out.println(highest.size());
break;
case "leave":
for (int j = 0; j < value; j++) {
Person left = stack.pop();
if (left == highest.peek()) {
highest.pop();
}
}
System.out.println(highest.size());
break;
}
}
}
关于Java栈队列获取可查看前台人数,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/48739747/