結果

問題 No.1890 Many Sequences Sum Queries
ユーザー ygd.ygd.
提出日時 2022-04-05 23:22:32
言語 Python3
(3.12.2 + numpy 1.26.4 + scipy 1.12.0)
結果
AC  
実行時間 664 ms / 2,000 ms
コード長 1,464 bytes
コンパイル時間 295 ms
コンパイル使用メモリ 12,672 KB
実行使用メモリ 10,880 KB
最終ジャッジ日時 2024-11-27 08:20:13
合計ジャッジ時間 5,434 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge1
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 664 ms
10,752 KB
testcase_01 AC 346 ms
10,880 KB
testcase_02 AC 28 ms
10,624 KB
testcase_03 AC 27 ms
10,496 KB
testcase_04 AC 27 ms
10,752 KB
testcase_05 AC 27 ms
10,752 KB
testcase_06 AC 28 ms
10,624 KB
testcase_07 AC 234 ms
10,880 KB
testcase_08 AC 65 ms
10,880 KB
testcase_09 AC 86 ms
10,880 KB
testcase_10 AC 262 ms
10,880 KB
testcase_11 AC 141 ms
10,880 KB
testcase_12 AC 420 ms
10,752 KB
testcase_13 AC 434 ms
10,752 KB
testcase_14 AC 255 ms
10,752 KB
testcase_15 AC 28 ms
10,496 KB
testcase_16 AC 28 ms
10,624 KB
testcase_17 AC 28 ms
10,624 KB
testcase_18 AC 28 ms
10,752 KB
testcase_19 AC 27 ms
10,624 KB
testcase_20 AC 27 ms
10,752 KB
testcase_21 AC 28 ms
10,752 KB
testcase_22 AC 26 ms
10,624 KB
testcase_23 AC 26 ms
10,624 KB
testcase_24 AC 26 ms
10,496 KB
testcase_25 AC 25 ms
10,624 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

import sys
#input = sys.stdin.readline
input = sys.stdin.buffer.readline #文字列はダメ
#sys.setrecursionlimit(1000000)
import bisect
#import itertools
#import random
#from heapq import heapify, heappop, heappush
#from collections import defaultdict 
#from collections import deque
#import copy
#import math
#from functools import lru_cache
#@lru_cache(maxsize=None)
#MOD = pow(10,9) + 7
#MOD = 998244353
#dx = [1,0,-1,0]
#dy = [0,1,0,-1]
#dx8 = [1,1,0,-1,-1,-1,0,1]
#dy8 = [0,1,1,1,0,-1,-1,-1]
#dx = [1,1,-1,-1]
#dy = [1,-1,1,-1]


def main():
    N,Q = map(int,input().split())
    A = list(map(int,input().split()))
    W = []
    for i,a in enumerate(A):
        wa = (1+a)*a//2
        W.append(wa)

    #print(W)
    SA = [0]
    for w in W:
        SA.append(SA[-1]+w)
    num = [0]
    for a in A:
        num.append(num[-1]+a)
    #print(SA)
    #print(num)
    #ret = []
    for _ in range(Q):
        s = int(input())
        if s > SA[-1]:
            print(-1)
            continue
        idx = bisect.bisect_right(SA,s)
        ans = num[idx-1]
        nokori = s - SA[idx-1]
        temp = calc(nokori)
        #print(ans,nokori)
        ans += temp
        print(ans)

def calc(v):
    if v == 0:
        return 0
    ng = 0
    ok = pow(10,5) + 100
    while abs(ok-ng) > 1:
        mid = (ok+ng)//2
        if (1+mid)*mid//2 >= v:
            ok = mid
        else:
            ng = mid
    return ok


if __name__ == '__main__':
    main()
0