結果

問題 No.3435 [Cherry 8th Tune *] 何回 LOVE SONG を書き換えただろうか?
コンテスト
ユーザー 👑 Kazun
提出日時 2024-09-25 01:07:37
言語 PyPy3
(7.3.17)
結果
AC  
実行時間 1,337 ms / 3,000 ms
コード長 864 bytes
記録
記録タグの例:
初AC ショートコード 純ショートコード 純主流ショートコード 最速実行時間
コンパイル時間 374 ms
コンパイル使用メモリ 82,208 KB
実行使用メモリ 150,016 KB
最終ジャッジ日時 2026-01-23 21:13:04
合計ジャッジ時間 46,303 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 48
権限があれば一括ダウンロードができます

ソースコード

diff #
raw source code 

def solve():
    # 入力
    A = input().strip()
    A_inv = { a: i for i, a in enumerate(A) }

    L = len(A)
    d = [None for _ in range(L)]
    for i in range(L):
        d[i] = list(map(int, input().split()))

    N, M = map(int, input().split())
    S = [None] * N
    for i in range(N):
        S[i] = [A_inv[x] for x in input().strip()]

    # 前計算
    F = [[0] * L for _ in range(M)]
    G = [None for _ in range(M)]
    for k in range(M):
        Fk = F[k]
        for i in range(N):
            Fk[S[i][k]] += 1

        G[k] = [sum(d[alpha][gamma] * Fk[gamma] for gamma in range(L)) for alpha in range(L)]

    # 本計算
    H = [sum(G[k][S[i][k]] for k in range(M)) for i in range(N)]

    return H

#==================================================
import sys
input = sys.stdin.readline
write = sys.stdout.write

print(*solve(), sep = "\n")
0