結果

問題 No.885 アマリクエリ
ユーザー 6soukiti296soukiti29
提出日時 2019-09-13 22:28:33
言語 Nim
(2.2.0)
結果
AC  
実行時間 1,146 ms / 2,000 ms
コード長 1,739 bytes
コンパイル時間 4,042 ms
コンパイル使用メモリ 66,472 KB
実行使用メモリ 19,200 KB
最終ジャッジ日時 2024-07-04 10:02:04
合計ジャッジ時間 9,477 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 19
権限があれば一括ダウンロードができます

ソースコード

diff # 

import sequtils,strutils,math
type
    priorityQue[I:static[int];T:tuple] = array[I + 1,T]
    item = tuple[cost,i : int]
proc push[T](A:var priorityQue; b: T;)=
    const key = 0
    A[0][0] += 1
    A[A[0][0]] = b # A[0][0]はsize、最大のIndex
    var index = A[0][0]
    while index > 1 and A[index] > A[index div 2]:
        (A[index],A[index div 2]) = (A[index div 2],A[index])
        index = index div 2
proc pop(A: var priorityQue):item =
    const key = 0
    result = A[1]
    A[1] = A[A[0][0]]
    var a : item
    A[A[0][0]] = a
    A[0][0] -= 1
    var index = 1
    while index * 2 <= A[0][0]:
        if index * 2 + 1 <= A[0][0]:
            if A[index] < A[index * 2] and A[index * 2 + 1] <= A[index * 2]:
                (A[index],A[index * 2]) = (A[index * 2],A[index])
                index = index * 2
            elif A[index] < A[index * 2 + 1]:
                (A[index],A[index * 2 + 1]) = (A[index * 2 + 1],A[index])
                index = index * 2 + 1
            else:
                break
        elif index * 2 <= A[0][0]:
            if A[index] < A[index * 2]:
                (A[index],A[index * 2]) = (A[index * 2],A[index])
            break
        else:
            break
proc size(A : priorityQue):int=
    return A[0][0]

var
    N = stdin.readline.parseInt
    A = stdin.readline.split.map(parseInt)
    Q = stdin.readline.parseInt
    X = stdin.readline.split.map(parseInt)
    pq : priorityQue[200010, item]
    v = newSeq[item](0)
    S = sum(A)

for i,a in A:
    pq.push((a,i))
    
for i,x in X:
    while pq.size > 0 and pq[1][0] >= x:
        var p = pq.pop
        S -= p.cost
        p.cost = p.cost mod x
        S += p.cost
        if p.cost > 0:
            pq.push(p)
    echo S
0