DMOPC '20 Contest 1 P2 - Victor's Moral Dilemma

View as PDF

Points: 5
Time limit: 0.6s
Memory limit: 128M

Author:
Problem type

Is killing an innocent person strictly wrong? ~ Victor, 2019

Victor has become obsessed with the Trolley Problem! Victor found the original trolley problem too boring, so he has devised his own version. In Victor's trolley problem, there is initially an array of trolleys, days, and the th trolley contains people. On the th day, Victor lines up all the remaining trolleys, and picks a number . He will then partition his array into two subarrays, and (where is the total number of trolleys on day ). If , then Victor will snap all the trolleys in out of existence and set equal to . Otherwise, he will snap all the trolleys in out of existence and set equal to .

Calculate the number of people Victor snaps on each day!

Constraints

• for all .
• The order of the trolleys will always remain the same.
• for all .

Input Specification

The first line will contain two space-separated integers and , denoting the initial number of trolleys and the number of days respectively.

The next line will contain space-separated integers , denoting the number of people in each trolley.

The next lines will each contain a single integer .

Output Specification

For each day, output the number of people that Victor will snap out of existence on a new line.

Sample Input

8 3
6 1 3 2 9 10 2 4
4
1
1

Sample Output

25
6
5

Explanation of Sample Output

On the first day, and . Then, and . Since , Victor will snap trolleys to out of existence, leaving as our array of trolleys.

On the second day, and . Then, and . Since , Victor will snap the first trolley and leave as our array.

On the third and last day, and . Then, and . Since , Victor will snap the last two trolleys and leave as our array.

• commented on Jan. 14, 2023, 9:33 p.m. edit 2

I'm still not sure why the brute force solution works given that and can be up to with up to . The fact that this is, at most, , makes sense. But when you plug numbers in, that's 30 million operations just in the iteration and summations. In the worst case scenario, a Python solution should take roughly 6 seconds, but mine finished in 0.2. Do the test cases not actually hit the constraint extremes? I am new to efficiency analysis so I'm sure I'm just missing something.

• commented on Oct. 13, 2020, 12:53 p.m. edited

It's too bad that brute force solutions passed during the contest, this was a nice prefix sum array problem.

• commented on Oct. 13, 2020, 1:01 p.m. edited

You can prove that the brute force solution only takes time, and was in fact intended.

• commented on Oct. 13, 2020, 6:12 p.m. edited

Interesting, if there was no upper bound on , would be worst case.

• commented on Oct. 13, 2020, 7:16 p.m.

Water is wet