結果
問題 |
No.1890 Many Sequences Sum Queries
|
ユーザー |
![]() |
提出日時 | 2022-12-03 22:07:49 |
言語 | Python3 (3.13.1 + numpy 2.2.1 + scipy 1.14.1) |
結果 |
AC
|
実行時間 | 1,222 ms / 2,000 ms |
コード長 | 776 bytes |
コンパイル時間 | 312 ms |
コンパイル使用メモリ | 12,544 KB |
実行使用メモリ | 10,880 KB |
最終ジャッジ日時 | 2024-10-11 01:19:11 |
合計ジャッジ時間 | 8,626 ms |
ジャッジサーバーID (参考情報) |
judge1 / judge4 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 1 |
other | AC * 25 |
ソースコード
from itertools import accumulate from bisect import bisect_left def main(): n, q = map(int, input().split()) a = [0] + list(map(int, input().split())) aa = [ai * (ai + 1) // 2 for ai in a] acc = list(accumulate(aa)) for _ in range(q): s = int(input()) if s > acc[-1]: print(-1) continue k = bisect_left(acc, s) if s == acc[k]: print(sum(a[:k + 1])) continue left = 0 right = a[k] + 1 while right - left > 1: mid = (right + left) // 2 if acc[k - 1] + mid * (mid + 1) // 2 >= s: right = mid else: left = mid print(sum(a[:k]) + right) if __name__ == "__main__": main()