Google KickStart 2020: Round F - ATM QUEUE (Editorial)

singhdevansh · September 27, 2020, 9:49am

ATM QUEUE was the first problem in the Round F of Google KickStart 2020, and here is my solution using python:
Solution using python

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	Output
2 3 3 2 7 4 5 6 9 10 4 7 2	Case #1: 1 3 2 Case #2: 3 5 1 2 4

In Sample Case #1, there are 3 people and the limit to withdraw in one turn is 3. Below is step-by-step description of how the process will look like:

The queue initially looks like [1, 2, 3]. The first person withdraws an amount of 2 in their first attempt and leaves the queue.
The queue now looks like [2, 3]. The second person wants to withdraw an amount of 7, but they can withdraw only 3 in their first turn. Since they still need to withdraw an amount of 4, they have to rejoin the queue at the end of the line.
The queue now looks like [3, 2]. The third person needs to withdraw an amount of 4 but they can only withdraw 3 in their first turn so, they rejoin the queue at the end of the line to withdraw amount of 1 later.
The queue now looks like [2, 3]. The second person still needs to withdraw an amount of 4. They withdraw an amount of 3 in their second turn and waits for their next turn to arrive to withdraw the remaining amount of 1.
The queue now looks like [3, 2]. The third person withdraws the remaining amount of 1 and leaves the queue.
The queue now looks like [2]. The second person withdraws the remaining amount of 1 and leaves the queue.
The queue is now empty.

The order in which people leave the queue is [1, 3, 2].

In Sample Case #2, there are 5 people and the limit to withdraw in one turn is 6. Below is step-by-step description of how the process will look like:

The queue initially looks like [1, 2, 3, 4, 5]. The first person withdraws an amount of 6, and joins at the end again to withdraw the remaining amount of 3 later.
The queue looks like [2, 3, 4, 5, 1]. The second person similarly withdraws an amount of 6 and waits for his next turn to withdraw an amount of 4.
The queue looks like [3, 4, 5, 1, 2]. The third person withdraws an amount of 4 and leaves the queue.
The queue now looks like [4, 5, 1, 2]. The fourth person withdraws 6 and waits for his next turn.
The queue looks like [5, 1, 2, 4]. The fifth person withdraws amount of 2 and leaves the queue.
The queue looks like, [1, 2, 4]. All other people now leave the queue after their second turn one by one.

The order in which people leave the queue is [3, 5, 1, 2, 4].

Code:

import math

for t in range (int(input())):
	n,x=map(int,input().split())
	arr=list(map(int,input().split()))
	ans=[]
	d=[]
	for i in range (n):
		d.append([arr[i],math.ceil(arr[i]/x),i])
	d.sort(key=lambda x: x[1])
	for i in d:
		ans.append(i[2]+1)
	print("Case #{}: ".format(t+1),*ans)

krishna_kc · September 27, 2020, 11:34am

add this too
Firstly, denote Ki as the number of times a person will use the ATM. Formally, Ki = ⌈ Ai / X ⌉.

Test Set 1

We can directly simulate the process using a queue.

Assume that i-th person, that wants to withdraw Ai , is first in the queue. There are two possibilites:

Ai ≤ X . In that case, this person withdraws Ai and leaves the queue. We can add i to the answer.
Ai > X . In that case, this person withdraws X (thus setting Ai to Ai - X ) and goes back to the end of the queue.

Time complexity of this simulation is O(Σ Ki).

In the worst case, when X = 1, Ki = Ai . Since Ai ≤ 100, the worst time complexity is O( N × 100), which easily fits into the time limit.

Test Set 2

In the second test set, Ki can be as big as 109, so direct simulation is too slow.

Let’s look at two people i and j. When will i-th person leave the queue before j-th person? There are two cases:

Ki < Kj. Since i-th person will use the ATM fewer times than j-th person, they will leave the queue earlier.
Ki = Kj and i < j. If they both use the ATM the same amount of times, the person earlier in the queue in the initial configuration will leave first.

This observation is enough to form a full solution. Sort people first in ascending order of Ki, and in case of ties in ascending order of their number. After sorting, this is our answer.

Time complexity of this solution is O( N log N ).

ffiq · September 27, 2020, 11:43am

lol

ghos3t · September 28, 2020, 1:42am

The problem analysis mentions that

Sort people first in ascending order of Ki, and in case of ties in ascending order of their number.

How does using the sort function used here ensure that in case of of ties for the arr[i]/x value the order of the queue number is used?

krishna_kc · September 28, 2020, 2:34am

you can use a custom comperator like this :

    if((p1.second-1)/cap == (p2.second-1)/cap){
        return p1.first<p2.first;
    }
    return p1.second/cap < p2.second/cap;
}

Here, I used this to sort a vector of pairs, the pair consists of (i, money he/she wants to withdraw)

singhdevansh · September 29, 2020, 3:52am

thanks for help!

harish1199 · October 17, 2020, 4:43am

Hi krishna_kc,

I used the same approach but still getting WA. Any idea wha I’m missing. Here’s my code:
#include <bits/stdc++.h>
using namespace std;
#define ll long long

int main() {
ll t;
cin>>t;
for(int k=0;k<t;k++) {
ll n, x;
cin>>n>>x;
ll a[n];
pair<ll,ll> p[n];
for(int i=0; i<n;i++) {
cin>>a[i];
}
for(int i=0; i<n;i++) {
p[i].first = (ceil(a[i]/x));
p[i].second = (i+1);
}
sort(p, p+n);
int testCase = k+1;
cout<< “Case #” << testCase << ": “;
for(int i=0;i<n;i++){
cout<<p[i].second;
if(i!=n-1) {
cout<<” ";
}
}
cout<<endl;
}
return 0;
}

Would appreciate if you can take a look.

krishna_kc · October 17, 2020, 5:57am

consider the following:

int a=4;
int b=5;
cout<<ceil(4/5);

output: 0

This is where the code went wrong.
both a and b are integers so the result was an integer , 0 which gives 0 when passed to ceiling function.
A small change will make your code work:

make this to

p[i].first = (ceil((double)a[i]/x));

this explicit conversion to double gives result in double so for the previous example of a/b,
the result of division will become 0.8 which becomes 1 by ceiling.
@harish1199