結果

問題 No.1513 simple 門松列 problem
コンテスト
ユーザー rin204
提出日時 2022-01-28 21:19:52
言語 PyPy3
(7.3.15)
結果
TLE  
実行時間 -
コード長 950 bytes
コンパイル時間 414 ms
コンパイル使用メモリ 82,176 KB
実行使用メモリ 211,840 KB
最終ジャッジ日時 2024-12-30 04:17:39
合計ジャッジ時間 25,996 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2 TLE * 1
other AC * 14 TLE * 4
権限があれば一括ダウンロードができます

ソースコード

diff # 

MOD = 998244353

n, k = map(int, input().split())
edges = [[] for _ in range(k * k)]
f = lambda x, y: x * k + y
dp = [0] * (k * k)
tot = [0] * (k * k)
for i in range(k):
    for j in range(i + 1, k):
        dp[f(i, j)] = 1
        tot[f(i, j)] = i + j
        for i2 in range(j):
            if i2 == i:
                continue
            edges[f(i, j)].append((f(j, i2)))

for i in range(k):
    for j in range(i):
        dp[f(i, j)] = 1
        tot[f(i, j)] = i + j
        for i2 in range(j + 1, k):
            if i2 == i:
                continue
            edges[f(i, j)].append((f(j, i2)))

for _ in range(n - 2):
    dp2 = [0] * (k * k)
    tot2 = [0] * (k * k)
    for pos in range(k * k):
        for npos in edges[pos]:
            dp2[npos] += dp[pos]
            dp2[npos] %= MOD
            tot2[npos] += tot[pos] + dp[pos] * (npos % k)
            tot2[npos] %= MOD
    dp = dp2
    tot = tot2
print(sum(dp) % MOD, sum(tot) % MOD)
0