結果

問題 No.3561 Collect KCPC
コンテスト
ユーザー detteiuu
提出日時 2026-05-31 04:47:13
言語 PyPy3
(7.3.17)
コンパイル:
pypy3 -mpy_compile _filename_
実行:
pypy3 _filename_
結果
AC  
実行時間 5,119 ms / 6,000 ms
コード長 1,650 bytes
記録
記録タグの例:
初AC ショートコード 純ショートコード 純主流ショートコード 最速実行時間
コンパイル時間 273 ms
コンパイル使用メモリ 85,888 KB
実行使用メモリ 145,632 KB
最終ジャッジ日時 2026-05-31 04:48:07
合計ジャッジ時間 45,917 ms
ジャッジサーバーID
（参考情報） 		judge1_0 / judge3_0
純コード判定待ち
このコードへのチャレンジ
（要ログイン）
サブタスク 配点 結果
部分点1 10 % AC * 15
部分点2 20 % AC * 15
部分点3 20 % AC * 13
部分点4 50 % AC * 51
合計 100 点
権限があれば一括ダウンロードができます

ソースコード

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

dist = [INF]*(N*12)
visited = [False]*(N*12)

def func(color):
    c = 0 if S[0] != "K" else 4
    for i in range(N*12):
        dist[i] = INF
        visited[i] = False
    dist[c] = 0
    que = [encode(0, 0, c)]
    while que:
        d, n, c = decode(heappop(que))
        if visited[n*12+c]:
            continue
        visited[n*12+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*12+c]+weight < dist[next*12+nc]:
                dist[next*12+nc] = dist[n*12+c]+weight
                heappush(que, encode(dist[next*12+nc], next, nc))
    return dist

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

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