結果

問題 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  
実行時間 562 ms / 2,000 ms
コード長 1,464 bytes
コンパイル時間 320 ms
コンパイル使用メモリ 10,932 KB
実行使用メモリ 8,540 KB
最終ジャッジ日時 2023-08-18 05:23:35
合計ジャッジ時間 5,122 ms
ジャッジサーバーID
（参考情報） 		judge15 / judge14
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 562 ms
8,328 KB
testcase_01 AC 281 ms
8,540 KB
testcase_02 AC 17 ms
8,232 KB
testcase_03 AC 16 ms
8,276 KB
testcase_04 AC 17 ms
8,140 KB
testcase_05 AC 17 ms
8,220 KB
testcase_06 AC 17 ms
8,288 KB
testcase_07 AC 193 ms
8,344 KB
testcase_08 AC 50 ms
8,360 KB
testcase_09 AC 61 ms
8,376 KB
testcase_10 AC 209 ms
8,524 KB
testcase_11 AC 113 ms
8,380 KB
testcase_12 AC 339 ms
8,456 KB
testcase_13 AC 357 ms
8,456 KB
testcase_14 AC 193 ms
8,468 KB
testcase_15 AC 17 ms
8,276 KB
testcase_16 AC 17 ms
8,276 KB
testcase_17 AC 18 ms
8,284 KB
testcase_18 AC 18 ms
8,340 KB
testcase_19 AC 17 ms
8,336 KB
testcase_20 AC 17 ms
8,228 KB
testcase_21 AC 16 ms
8,336 KB
testcase_22 AC 17 ms
8,336 KB
testcase_23 AC 17 ms
8,164 KB
testcase_24 AC 17 ms
8,236 KB
testcase_25 AC 17 ms
8,260 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