Submission #68130460


Source Code Expand

from heapq import heapify, heappop, heappush


N=int(input())
PAB=[]
for _ in range(N):
    p,a,b=map(int,input().split())
    PAB.append((p,a,b))

Q=int(input())
X=[(int(input()),i) for i in range(Q)]

heapify(X)
SIZE=1000
pos=[[] for _ in range(SIZE+1)]

total=0
while X:
    x,i=heappop(X)
    if x+total<=SIZE:
        pos[x+total].append(i)
    else:
        heappush(X,(x,i))
        break

for p,a,b in PAB:
    posn=[[] for _ in range(SIZE+1)]
    for t in range(p+1):
        tn=a+t
        if len(pos[t])>len(posn[tn]):
            pos[t],posn[tn]=posn[tn],pos[t]
        posn[tn].extend(pos[t])

    for t in range(p+1,SIZE+1):
        tn=max(0,t-b)
        if len(pos[t])>len(posn[tn]):
            pos[t],posn[tn]=posn[tn],pos[t]
        posn[tn].extend(pos[t])
    
    total-=b
    while X:
        x,i=heappop(X)
        if x+total<=SIZE:
            posn[x+total].append(i)
        else:
            heappush(X,(x,i))
            break
    pos=posn

ans=[-1]*Q
for t,line in enumerate(pos):
    for i in line:
        ans[i]=t
for x,i in X:
    ans[i]=max(0,x+total)
print(*ans,sep="\n")

Submission Info

Submission Time
Task D - Takahashi's Expectation
User mo12412
Language Python (PyPy 3.10-v7.3.12)
Score 425
Code Size 1159 Byte
Status AC
Exec Time 575 ms
Memory 126856 KiB

Judge Result

Set Name Sample All
Score / Max Score 0 / 0 425 / 425
Status
AC × 3
AC × 30
Set Name Test Cases
Sample 00_sample_00.txt, 00_sample_01.txt, 00_sample_02.txt
All 00_sample_00.txt, 00_sample_01.txt, 00_sample_02.txt, 01_random_03.txt, 01_random_04.txt, 01_random_05.txt, 01_random_06.txt, 01_random_07.txt, 01_random_08.txt, 01_random_09.txt, 01_random_10.txt, 01_random_11.txt, 01_random_12.txt, 01_random_13.txt, 01_random_14.txt, 01_random_15.txt, 01_random_16.txt, 01_random_17.txt, 01_random_18.txt, 01_random_19.txt, 01_random_20.txt, 01_random_21.txt, 01_random_22.txt, 01_random_23.txt, 01_random_24.txt, 01_random_25.txt, 01_random_26.txt, 01_random_27.txt, 01_random_28.txt, 01_random_29.txt
Case Name Status Exec Time Memory
00_sample_00.txt AC 62 ms 81416 KiB
00_sample_01.txt AC 62 ms 81036 KiB
00_sample_02.txt AC 71 ms 81308 KiB
01_random_03.txt AC 443 ms 126564 KiB
01_random_04.txt AC 473 ms 125804 KiB
01_random_05.txt AC 443 ms 126364 KiB
01_random_06.txt AC 453 ms 125672 KiB
01_random_07.txt AC 448 ms 125868 KiB
01_random_08.txt AC 457 ms 126232 KiB
01_random_09.txt AC 456 ms 126272 KiB
01_random_10.txt AC 445 ms 126168 KiB
01_random_11.txt AC 459 ms 126856 KiB
01_random_12.txt AC 433 ms 126036 KiB
01_random_13.txt AC 443 ms 125844 KiB
01_random_14.txt AC 443 ms 125684 KiB
01_random_15.txt AC 457 ms 125908 KiB
01_random_16.txt AC 246 ms 97988 KiB
01_random_17.txt AC 196 ms 99928 KiB
01_random_18.txt AC 405 ms 123460 KiB
01_random_19.txt AC 209 ms 91888 KiB
01_random_20.txt AC 241 ms 94732 KiB
01_random_21.txt AC 381 ms 123776 KiB
01_random_22.txt AC 323 ms 123436 KiB
01_random_23.txt AC 248 ms 106588 KiB
01_random_24.txt AC 408 ms 124952 KiB
01_random_25.txt AC 488 ms 125408 KiB
01_random_26.txt AC 431 ms 124960 KiB
01_random_27.txt AC 498 ms 125500 KiB
01_random_28.txt AC 433 ms 125708 KiB
01_random_29.txt AC 575 ms 125612 KiB