結果

問題 No.3561 Collect KCPC
コンテスト
ユーザー detteiuu
提出日時 2026-05-31 04:35:02
言語 PyPy3
(7.3.17)
コンパイル:
pypy3 -mpy_compile _filename_
実行:
pypy3 _filename_
結果
TLE  
実行時間 -
コード長 1,568 bytes
記録
記録タグの例:
初AC ショートコード 純ショートコード 純主流ショートコード 最速実行時間
コンパイル時間 325 ms
コンパイル使用メモリ 85,376 KB
実行使用メモリ 337,408 KB
最終ジャッジ日時 2026-05-31 04:37:54
合計ジャッジ時間 53,575 ms
ジャッジサーバーID
（参考情報） 		judge2_0 / judge1_1
このコードへのチャレンジ
（要ログイン）
サブタスク 配点 結果
部分点1 10 % AC * 15
部分点2 20 % AC * 15
部分点3 20 % AC * 13
部分点4 50 % AC * 44 TLE * 1 -- * 6
合計 50 点
権限があれば一括ダウンロードができます

ソースコード

diff #
raw source code 

from sys import stdin
input = stdin.readline
from random import randrange
from heapq import heappush, heappop

base1, base2 = 10**5*12, 12
def encode(d, n, c):
    return d*base1+n*base2+c

base3 = 10**5
def decode(n):
    return n//base1, n//base2%base3, n%base2

N, M = map(int, input().split())
G = [[] for _ in range(N)]
for _ in range(M):
    u, v, c = map(int, input().split())
    u, v = u-1, v-1
    G[u].append((v, c))
S = input().rstrip("\n")

INF = 1<<60

def func(color):
    c = 0 if S[0] != "K" else 4
    dist = [[INF]*12 for _ in range(N)]
    dist[0][c] = 0
    visited = [[False]*12 for _ in range(N)]
    que = [encode(0, 0, c)]
    while que:
        d, n, c = decode(heappop(que))
        if visited[n][c]:
            continue
        visited[n][c] = True
        for next, weight in G[n]:
            nc = c
            if nc//4 == 0 and S[next] == "K":
                nc += 4
            elif nc//4 == 1 and S[next] == "C" and nc%4 == 0:
                nc |= 1<<color[next]
            elif nc//4 == 1 and 1 <= nc%4 <= 2 and S[next] == "P":
                nc += 4
            elif nc//4 == 2 and c>>color[next] & 1 == 0 and S[next] == "C":
                nc |= 1<<color[next]
            if dist[n][c]+weight < dist[next][nc]:
                dist[next][nc] = dist[n][c]+weight
                heappush(que, encode(dist[next][nc], next, nc))
    return dist

ans = INF
for _ in range(7):
    dist = func([randrange(2) for _ in range(N)])
    MIN = min(dist[i][-1] for i in range(N))
    ans = min(ans, MIN)

print(ans if ans != INF else -1)
0