所以有測試資料可以在
https://github.com/algo-seacow/code/tree/master/apcs/10907
找到。
如果有希望可以加強的測試資料可以 email 到 algo.seacow@gmail.com 跟本題的管理者聯絡,或是直接在 github 發 PR。
您好,請問我這Python code只有20分,其餘TLE,有辦法再優化時間嗎? 謝謝。
[n,m]=[int(x) for x in input().split()]
p=[int(x) for x in input().split()]
pp=p+p
q=[int(x) for x in input().split()]
spp=[0]*(2*n)
spp[0]=pp[0]
for i in range(1,len(pp)):
spp[i]=spp[i-1]+pp[i]
a=0 # a is house index as one mission finished, initially at 0
for i in range(len(q)):
L=a # left index at initial
R=a+n-1 # right index at initial
while L<R:
m=int((L+R)/2)
if q[i]>spp[m]-spp[a]+pp[a]:
L=m+1
else: R=m
a=(L+1)%n # L=R
print(a)