結果
問題 | No.1890 Many Sequences Sum Queries |
ユーザー | DrDrpilot |
提出日時 | 2022-06-09 13:12:11 |
言語 | Python3 (3.12.2 + numpy 1.26.4 + scipy 1.12.0) |
結果 |
AC
|
実行時間 | 1,346 ms / 2,000 ms |
コード長 | 783 bytes |
コンパイル時間 | 103 ms |
コンパイル使用メモリ | 11,912 KB |
実行使用メモリ | 10,248 KB |
最終ジャッジ日時 | 2023-10-21 04:28:46 |
合計ジャッジ時間 | 8,815 ms |
ジャッジサーバーID (参考情報) |
judge12 / judge9 |
(要ログイン)
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 1,346 ms
10,228 KB |
testcase_01 | AC | 758 ms
10,248 KB |
testcase_02 | AC | 31 ms
10,212 KB |
testcase_03 | AC | 30 ms
10,212 KB |
testcase_04 | AC | 31 ms
10,212 KB |
testcase_05 | AC | 30 ms
10,212 KB |
testcase_06 | AC | 31 ms
10,212 KB |
testcase_07 | AC | 415 ms
10,228 KB |
testcase_08 | AC | 101 ms
10,236 KB |
testcase_09 | AC | 124 ms
10,236 KB |
testcase_10 | AC | 706 ms
10,228 KB |
testcase_11 | AC | 354 ms
10,228 KB |
testcase_12 | AC | 753 ms
10,228 KB |
testcase_13 | AC | 796 ms
10,228 KB |
testcase_14 | AC | 439 ms
10,228 KB |
testcase_15 | AC | 30 ms
10,212 KB |
testcase_16 | AC | 30 ms
10,212 KB |
testcase_17 | AC | 30 ms
10,212 KB |
testcase_18 | AC | 30 ms
10,212 KB |
testcase_19 | AC | 29 ms
10,212 KB |
testcase_20 | AC | 30 ms
10,212 KB |
testcase_21 | AC | 29 ms
10,212 KB |
testcase_22 | AC | 31 ms
10,212 KB |
testcase_23 | AC | 31 ms
10,212 KB |
testcase_24 | AC | 30 ms
10,212 KB |
testcase_25 | AC | 30 ms
10,212 KB |
ソースコード
import bisect from itertools import accumulate n,q=map(int,input().split()) a=list(map(int,input().split())) a_ind=list(accumulate(a)) b=[] for i in a: b.append(i*(i+1)//2) b=list(accumulate(b)) for _ in range(q): s=int(input()) if s>b[-1]: print(-1) continue ind=bisect.bisect_right(b,s) if ind==0: ok=a[0];ng=0 while ok-ng>1: mid=(ok+ng)//2 if mid*(mid+1)//2>=s: ok=mid else: ng=mid print(ok) continue s-=b[ind-1] if s==0: print(a_ind[ind-1]) continue ok=a[ind];ng=0 while ok-ng>1: mid=(ok+ng)//2 if mid*(mid+1)//2>=s: ok=mid else: ng=mid print(a_ind[ind-1]+ok)