Java栈队列获取可查看前台人数

Question

我有一个模拟队列的程序，其中有两种类型的输入 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 对象来存储最高的人变量，但是有没有更优雅的方法来解决这个问题？

Answer 1

我认为您可以使用 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;
        }
    }
}