結果

問題 No.1029 JJOOII 3
ユーザー maspy
提出日時 2020-04-17 23:37:14
言語 Python3
(3.13.1 + numpy 2.2.1 + scipy 1.14.1)
結果
WA  
実行時間 -
コード長 1,054 bytes
コンパイル時間 411 ms
コンパイル使用メモリ 12,800 KB
実行使用メモリ 13,312 KB
最終ジャッジ日時 2024-10-03 15:35:20
合計ジャッジ時間 3,151 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 16 WA * 22
権限があれば一括ダウンロードができます

ソースコード

diff # 

import sys
read = sys.stdin.buffer.read
readline = sys.stdin.buffer.readline
readlines = sys.stdin.buffer.readlines

N, K = map(int, readline().split())
S = []
C = []
for _ in range(N):
    x, y = readline().split()
    x = tuple(0 if x == 'J' else 1 if x == 'O' else 2 for x in x.decode())
    S.append(x)
    C.append(int(y))

INF = 10 ** 16
dp = [INF] * (K + K + K + 1)
dp[0] = 0

cnt = [0] * (N * 3)
for q in range(3):
    for i in range(N):
        cnt[3 * i + q] = sum(n == q for n in S[i])


def solve_small():
    def f(k, i):
        q, r = divmod(k, K)
        if k + cnt[3 * i + q] < K:
            return k + cnt[3 * i + q]
        for x in S[i]:
            if x == q:
                r += 1
                if r == K:
                    r = 0
                    q += 1
        return K * q + r

    for k in range(K + K + K):
        for i in range(N):
            next_k = f(k, i)
            dp[next_k] = min(dp[k] + C[i], dp[next_k])

    x = dp[-1]
    if x == INF:
        x = -1
    return x


if K <= 100:
    print(solve_small())
0