結果

問題 No.762 PDCAパス
ユーザー tails1434
提出日時 2020-01-15 07:25:33
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 246 ms / 2,000 ms
コード長 1,279 bytes
コンパイル時間 1,203 ms
コンパイル使用メモリ 81,792 KB
実行使用メモリ 88,448 KB
最終ジャッジ日時 2024-12-30 13:45:55
合計ジャッジ時間 8,340 ms
ジャッジサーバーID
(参考情報)
judge1 / judge5
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 3
other AC * 38
権限があれば一括ダウンロードができます

ソースコード

diff #

import sys
input = sys.stdin.readline

def main():
    N, M = map(int, input().split())
    S = input()
    d = [[] for _ in range(N)]
    MOD = 10 ** 9 + 7
    for _ in range(M):
        u, v = map(int, input().split())
        u -= 1
        v -= 1
        if S[u] == 'P' and S[v] == 'D':
            d[u].append(v)
        if S[u] == 'D' and S[v] == 'C':
            d[u].append(v)
        if S[u] == 'C' and S[v] == 'A':
            d[u].append(v)
        if S[v] == 'P' and S[u] == 'D':
            d[v].append(u)
        if S[v] == 'D' and S[u] == 'C':
            d[v].append(u)
        if S[v] == 'C' and S[u] == 'A':
            d[v].append(u)
    
    dp = [[0] * N for _ in range(4)]

    for i in range(N):
        if S[i] == 'P':
            dp[0][i] = 1
    
    for u in range(N):
        for v in d[u]:
            if S[v] == 'D':
                dp[1][v] += dp[0][u]
                dp[1][v] %= MOD
    
    for u in range(N):
        for v in d[u]:
            if S[v] == 'C':
                dp[2][v] += dp[1][u]
                dp[2][v] %= MOD
    
    for u in range(N):
        for v in d[u]:
            if S[v] == 'A':
                dp[3][v] += dp[2][u]
                dp[3][v] %= MOD

    print(sum(dp[3]) % MOD)

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