結果

問題 No.1890 Many Sequences Sum Queries
ユーザー かもめのうで
提出日時 2023-09-26 12:39:22
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 270 ms / 2,000 ms
コード長 502 bytes
コンパイル時間 371 ms
コンパイル使用メモリ 81,900 KB
実行使用メモリ 88,556 KB
最終ジャッジ日時 2024-07-19 07:05:32
合計ジャッジ時間 3,915 ms
ジャッジサーバーID
(参考情報)
judge3 / judge5
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 1
other AC * 25
権限があれば一括ダウンロードができます

ソースコード

diff #

from itertools import accumulate
n, q = map(int, input().split())
a = list(map(int, input().split()))
b = []
for i in a:
    for j in range(1, i+1):
        b.append(j)
# print(b)
rui = list(accumulate(b, initial=0))
# print(rui)

for i in range(q):
    x = int(input())
    ng = -1
    ok = len(rui)
    while abs(ok-ng) > 1:
        mid = (ok+ng)//2
        if rui[mid] >= x:
            ok = mid
        else:
            ng = mid
    if ok == len(rui):
        print(-1)
    else:
        print(ok)
0