【Kick Start】ATM Queue

题目思路

• 这题的思路是从前往后一次遍历，判断出每个人需要排队几次才可以取完钱。 按照排队的次数，有序按照index排列
• 一开始想到用hashmap存储，key是排队次数，value是人员编号的list
  但是前一阵还总结了TreeMap是一种按照key有序存储的结构，所以这里用TreeMap，key是排队次数，value是人员编号的list
• 队次数应该用(要取的钱 - 1) / 次可取的最大钱数，这里一定要减一，如果不减一的话，当要取的钱等于可取的最大钱数的时候，排队的次数就是1次了，其实应该是0次。
• 最后遍历一遍TreeMap

题目描述

原文
Problem
There are N people numbered from 1 to N, standing in a queue to withdraw money from an ATM. The queue is formed in ascending order of their number. The person numbered i wants to withdraw amount Ai. The maximum amount a person can withdraw at a time is X. If they need more money than X, they need to go stand at the end of the queue and wait for their turn in line. A person leaves the queue once they have withdrawn the required amount.You need to find the order in which all the people leave the queue.
Input
The first line of the input gives the number of test cases T. T test cases follow.The first line of each test case gives two space separated integers: the number of people standing in the queue, N and the maximum amount X that can be withdrawn in one turn.The next line contains N space separated integers Ai.
Output
For each test case, output one line containing Case #x: y, where x is the test case number (starting from 1) and y is the space separated list of integers that denote the order in which the people leave the queue.
Limits
Time limit: 20 seconds per test set.
Memory limit: 1GB.
1 ≤ T ≤ 100.
Test Set 1
1 ≤ N ≤ 100.
1 ≤ Ai ≤ 100.
1 ≤ X ≤ 100.
Test Set 2
1 ≤ N ≤ 105 for at most 10 test cases. For the remaining cases, 1 ≤ N ≤ 100
1 ≤ Ai ≤ 109.
1 ≤ X ≤ 109.
Sample
Input
2
3 3
2 7 4
5 6
9 10 4 7 2
Output
Case #1: 1 3 2
Case #2: 3 5 1 2 4

代码

import java.util.*;
import java.io.*;
public class Solution {
    public static void main(String[] args) {
        Scanner in = new Scanner(new BufferedReader(new InputStreamReader(System.in)));
        int t = in.nextInt(); // Scanner has functions to read ints, longs, strings, chars, etc.
        for (int i = 1; i <= t; ++i) {
            TreeMap<Integer, LinkedList<Integer>> temp = new TreeMap<>();
            int people = in.nextInt();
            int money = in.nextInt();
            for(int j = 1; j <= people; ++j){
                temp.computeIfAbsent((in.nextInt() - 1) / money, k -> new LinkedList<>()).addLast(j);
            }
            StringBuilder str = new StringBuilder();
            for(LinkedList<Integer> list: temp.values()){
                for (int item : list){
                    str.append(item).append(" ");
                }
            }
            String res = str.toString();
            res = res.substring(0, res.length() - 1);
            System.out.println("Case #" + i + ": " + res);
        }
    }
}