結果

問題 No.762 PDCAパス
ユーザー hiragnhiragn
提出日時 2022-11-29 07:55:09
言語 Python3
(3.13.1 + numpy 2.2.1 + scipy 1.14.1)
結果
AC  
実行時間 557 ms / 2,000 ms
コード長 950 bytes
コンパイル時間 185 ms
コンパイル使用メモリ 12,800 KB
実行使用メモリ 34,724 KB
最終ジャッジ日時 2024-10-06 15:19:25
合計ジャッジ時間 11,498 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 38
権限があれば一括ダウンロードができます

ソースコード

diff # 

from collections import defaultdict


def main():
    mod = 10 ** 9 + 7
    n, m = map(int, input().split())
    s = input()

    adj = defaultdict(list)
    for _ in range(m):
        a, b = map(int, input().split())
        a -= 1
        b -= 1
        if s[a] + s[b] in ["PD", "DC", "CA"]:
            adj[a].append(b)
        if s[b] + s[a] in ["PD", "DC", "CA"]:
            adj[b].append(a)

    wd = "PDCA"
    # dp[i][v]:PDCAのi文字目まで来ていて，終点が頂点vの経路数
    dp = [[0 for _ in range(n)] for _ in range(4)]

    # 1文字目
    for i in range(n):
        if s[i] == wd[0]:
            dp[0][i] = 1

    # 2文字目以降
    for i in range(1, len(wd)):
        for u in range(n):
            for v in adj[u]:
                if s[v] == wd[i]:
                    dp[i][v] += dp[i - 1][u]
                    dp[i][v] %= mod

    res = sum(dp[3]) % mod
    print(res)


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