結果

問題 No.2310 [Cherry 5th Tune A] Against Regret
ユーザー 👑 Kazun
提出日時 2023-02-09 06:19:18
言語 PyPy3
(7.3.15)
結果
TLE  
実行時間 -
コード長 1,174 bytes
コンパイル時間 399 ms
コンパイル使用メモリ 82,304 KB
実行使用メモリ 81,408 KB
最終ジャッジ日時 2024-09-15 21:46:28
合計ジャッジ時間 9,950 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 3 TLE * 1 -- * 26
権限があれば一括ダウンロードができます

ソースコード

diff # 

def solve():
    from itertools import permutations

    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())

        for L in range(K+1):
            for P in permutations(range(K),r=L):
                temp=1; prev=0
                for p in P:
                    temp*=(S[prev][A[p]]*C[p])%Mod; temp%=Mod
                    prev=B[p]

                temp*=S[prev][N]; temp%=Mod
                Ans[q]+=temp

        Ans[q]%=Mod

    return Ans

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

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