結果

問題 No.1513 simple 門松列 problem
コンテスト
ユーザー convexineq
提出日時 2021-05-21 23:19:43
言語 PyPy3
(7.3.15)
結果
TLE  
実行時間 -
コード長 979 bytes
コンパイル時間 236 ms
コンパイル使用メモリ 82,008 KB
実行使用メモリ 87,020 KB
最終ジャッジ日時 2024-10-10 10:11:55
合計ジャッジ時間 4,822 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2 TLE * 1
other -- * 18
権限があれば一括ダウンロードができます

ソースコード

diff # 

n,k = map(int,input().split())
MOD = 998244353

dpn = [[0]*k for _ in range(k)]
dps = [[0]*k for _ in range(k)]
for i in range(k):
    for j in range(i+1,k):
            dpn[i][j] = dpn[j][i] = 1
            dps[i][j] = dps[j][i] = i+j

for I in range(n-2):
    ndpn = [[0]*k for _ in range(k)]
    ndps = [[0]*k for _ in range(k)]
    for i in range(k):
        for j in range(i+1,k):
            for p in range(j):
                if i==p: continue
                ndpn[j][p] += dpn[i][j]
                ndpn[j][p] %= MOD
                ndps[j][p] += dps[i][j] + dpn[i][j]*p
                ndps[j][p] %= MOD
        for j in range(i):
            for p in range(j+1,k):
                if i==p: continue
                ndpn[j][p] += dpn[i][j]
                ndpn[j][p] %= MOD
                ndps[j][p] += dps[i][j] + dpn[i][j]*p
                ndps[j][p] %= MOD

    dpn = ndpn
    dps = ndps

v = sum(sum(i) for i in dpn)%MOD
w = sum(sum(i) for i in dps)%MOD
print(v,w)
0