結果
問題 | No.1513 simple 門松列 problem |
ユーザー | convexineq |
提出日時 | 2021-05-21 23:32:55 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 423 ms / 3,000 ms |
コード長 | 1,128 bytes |
コンパイル時間 | 644 ms |
コンパイル使用メモリ | 82,844 KB |
実行使用メモリ | 79,620 KB |
最終ジャッジ日時 | 2024-04-18 16:58:42 |
合計ジャッジ時間 | 4,636 ms |
ジャッジサーバーID (参考情報) |
judge3 / judge1 |
(要ログイン)
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 38 ms
52,736 KB |
testcase_01 | AC | 41 ms
52,224 KB |
testcase_02 | AC | 423 ms
79,428 KB |
testcase_03 | AC | 50 ms
61,440 KB |
testcase_04 | AC | 39 ms
52,352 KB |
testcase_05 | AC | 54 ms
64,384 KB |
testcase_06 | AC | 54 ms
64,512 KB |
testcase_07 | AC | 46 ms
60,288 KB |
testcase_08 | AC | 66 ms
69,120 KB |
testcase_09 | AC | 38 ms
52,224 KB |
testcase_10 | AC | 39 ms
52,864 KB |
testcase_11 | AC | 38 ms
52,352 KB |
testcase_12 | AC | 38 ms
52,224 KB |
testcase_13 | AC | 38 ms
52,864 KB |
testcase_14 | AC | 416 ms
79,368 KB |
testcase_15 | AC | 420 ms
79,620 KB |
testcase_16 | AC | 423 ms
79,232 KB |
testcase_17 | AC | 294 ms
78,008 KB |
testcase_18 | AC | 156 ms
77,952 KB |
testcase_19 | AC | 71 ms
70,912 KB |
testcase_20 | AC | 125 ms
77,056 KB |
ソースコード
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 _ in range(n-2): ndpn = [[0]*(k+1) for _ in range(k)] for i in range(k): for j in range(i+1,k): ndpn[j][0] += dpn[i][j] ndpn[j][0] %= MOD ndpn[j][i] -= dpn[i][j] ndpn[j][i] %= MOD ndpn[j][i+1] += dpn[i][j] ndpn[j][i+1] %= MOD ndpn[j][j] -= dpn[i][j] ndpn[j][j] %= MOD for j in range(i): ndpn[j][j+1] += dpn[i][j] ndpn[j][j+1] %= MOD ndpn[j][i] -= dpn[i][j] ndpn[j][i] %= MOD ndpn[j][i+1] += dpn[i][j] ndpn[j][i+1] %= MOD ndpn[j][k] -= dpn[i][j] ndpn[j][k] %= MOD dpn = ndpn for i in range(k): for j in range(1,k+1): dpn[i][j] += dpn[i][j-1] dpn[i][j] %= MOD v = sum(sum(i) for i in dpn)%MOD w = v*(k-1)*n*(MOD+1)//2%MOD print(v,w)