結果

問題 No.2310 [Cherry 5th Tune A] Against Regret
コンテスト
ユーザー 👑 Kazun
提出日時 2023-03-04 15:37:31
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 2,134 ms / 6,000 ms
コード長 1,124 bytes
コンパイル時間 421 ms
コンパイル使用メモリ 82,176 KB
実行使用メモリ 80,768 KB
最終ジャッジ日時 2024-12-16 08:47:48
合計ジャッジ時間 34,348 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 30
権限があれば一括ダウンロードができます

ソースコード

diff # 

def solve():
    N=int(input())

    X=[None]*(N+1)
    for u in range(N+1):
        X[u]=list(map(int,input().split()))

    Mod=998244353

    # 前計算
    S=[[0]*(N+1) for _ in range(N+1)]
    for u in range(N+1):
        Su=S[u]
        Su[u]=1
        for v in range(u,N+1):
            Xv=X[v]
            for w in range(v+1,N+1):
                Su[w]+=Su[v]*Xv[w]
                Su[w]%=Mod

    Q=int(input())
    Ans=[0]*Q
    for q in range(Q):
        K=int(input())

        A=[0]*K; B=[0]*K; C=[0]*K

        for k in range(K):
            A[k],B[k],C[k]=map(int,input().split())

        DP=[0]*K
        for k in range(K):
            DP[k]=S[0][A[k]]

        Ans[q]=S[0][N]
        for k in sorted(range(K), key=lambda k:A[k]):
            for l in range(K):
                alpha=C[k]*S[B[k]][A[l]]%Mod
                DP[l]+=DP[k]*alpha
                DP[l]%=Mod

            Ans[q]+=DP[k]*(C[k]*S[B[k]][N]%Mod)
            Ans[q]%=Mod

    return Ans

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

write("\n".join(map(str,solve())))
0