結果

問題 No.1288 yuki collection
コンテスト
ユーザー LyricalMaestro
提出日時 2024-10-12 14:55:57
言語 PyPy3
(7.3.15)
結果
WA  
実行時間 -
コード長 894 bytes
記録
記録タグの例:
初AC ショートコード 純ショートコード 純主流ショートコード 最速実行時間
コンパイル時間 163 ms
コンパイル使用メモリ 82,436 KB
実行使用メモリ 74,980 KB
最終ジャッジ日時 2024-10-12 14:56:02
合計ジャッジ時間 3,896 ms
ジャッジサーバーID
（参考情報） 		judge / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 10 WA * 30
権限があれば一括ダウンロードができます

ソースコード

diff #
raw source code 

## https://yukicoder.me/problems/no/1288

import heapq

def main():
    N = int(input())
    S = input()
    V = list(map(int, input().split()))


    i_queue = []
    k_queue = []
    u_queue = []
    answer = 0
    for i in reversed(range(N)):
        s = S[i]
        if s == "i":
            heapq.heappush(i_queue, -V[i])
        elif s == "k":
            if len(i_queue) > 0:
                v = heapq.heappop(i_queue)
                v = -v
                heapq.heappush(k_queue, -(V[i] + v))
        elif s == "u":
            if len(k_queue) > 0:
                v = heapq.heappop(k_queue)
                v = -v
                heapq.heappush(u_queue, -(V[i] + v))
        else:
            if len(u_queue) > 0:
                v = heapq.heappop(u_queue)
                v = -v
                answer += v + V[i]
    print(answer)
            

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