結果

問題 No.2310 [Cherry 5th Tune A] Against Regret
コンテスト
ユーザー mkawa2
提出日時 2023-05-22 16:39:02
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 1,745 ms / 6,000 ms
コード長 1,016 bytes
コンパイル時間 334 ms
コンパイル使用メモリ 81,792 KB
実行使用メモリ 82,296 KB
最終ジャッジ日時 2024-12-22 12:24:21
合計ジャッジ時間 32,271 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 30
権限があれば一括ダウンロードができます

ソースコード

diff #
raw source code 

mod = 998244353

n = int(input())
x = [list(map(int, input().split())) for _ in range(n+1)]

# 初期の辺だけでの
pre = [[0]*(n+1) for _ in range(n+1)]
for i in range(n+1):
    pre[i][i] = 1
    for j in range(i+1, n+1):
        for k in range(i, j):
            pre[i][j] += pre[i][k]*x[k][j]%mod
            pre[i][j] %= mod

q = int(input())
for _ in range(q):
    k = int(input())
    abc = [list(map(int, input().split())) for _ in range(k)]
    abc.sort(key=lambda x: x[1])

    # dp[i]...最後に使った辺がabc[i]のときのパスの数
    dp = [0]*k
    # 追加の辺を使わない場合を予め答えに足しておく
    ans = pre[0][n]
    for i in range(k):
        a, b, c = abc[i]
        dp[i] = pre[0][a]
        for j in range(i):
            p = abc[j][1]
            dp[i] += dp[j]*pre[p][a]%mod
        dp[i] *= c
        dp[i] %= mod
        # これ以上追加の辺を使わない場合を答えに足す
        ans += dp[i]*pre[b][n]%mod
        ans %= mod

    print(ans)
0