結果
問題 | No.1513 simple 門松列 problem |
ユーザー | ayaoni |
提出日時 | 2021-05-22 05:57:43 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 605 ms / 3,000 ms |
コード長 | 1,564 bytes |
コンパイル時間 | 142 ms |
コンパイル使用メモリ | 82,260 KB |
実行使用メモリ | 205,056 KB |
最終ジャッジ日時 | 2024-04-18 17:22:17 |
合計ジャッジ時間 | 4,655 ms |
ジャッジサーバーID (参考情報) |
judge3 / judge2 |
(要ログイン)
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 39 ms
52,352 KB |
testcase_01 | AC | 37 ms
52,992 KB |
testcase_02 | AC | 605 ms
205,056 KB |
testcase_03 | AC | 46 ms
62,464 KB |
testcase_04 | AC | 37 ms
53,248 KB |
testcase_05 | AC | 49 ms
64,000 KB |
testcase_06 | AC | 53 ms
64,896 KB |
testcase_07 | AC | 44 ms
61,952 KB |
testcase_08 | AC | 54 ms
66,560 KB |
testcase_09 | AC | 37 ms
52,608 KB |
testcase_10 | AC | 37 ms
52,864 KB |
testcase_11 | AC | 36 ms
52,864 KB |
testcase_12 | AC | 36 ms
52,736 KB |
testcase_13 | AC | 36 ms
52,736 KB |
testcase_14 | AC | 579 ms
204,288 KB |
testcase_15 | AC | 598 ms
204,236 KB |
testcase_16 | AC | 585 ms
204,876 KB |
testcase_17 | AC | 455 ms
159,800 KB |
testcase_18 | AC | 201 ms
104,912 KB |
testcase_19 | AC | 62 ms
70,656 KB |
testcase_20 | AC | 135 ms
91,304 KB |
ソースコード
import sys sys.setrecursionlimit(10**7) def I(): return int(sys.stdin.readline().rstrip()) def MI(): return map(int,sys.stdin.readline().rstrip().split()) def LI(): return list(map(int,sys.stdin.readline().rstrip().split())) def LI2(): return list(map(int,sys.stdin.readline().rstrip())) def S(): return sys.stdin.readline().rstrip() def LS(): return list(sys.stdin.readline().rstrip().split()) def LS2(): return list(sys.stdin.readline().rstrip()) N,K = MI() mod = 998244353 dp = [[[0]*K for _ in range(K)] for _ in range(N+1)] # dp[i][j][k] = 長さiで最後がk,jである門松列列の個数 S_dp = [[[0]*K for _ in range(K)] for _ in range(N+1)] # S_dp[i][j][k] = dp[i][j][0]+…+dp[i][j][k] for j in range(K): for k in range(K): if j != k: dp[2][j][k] = 1 if k == 0: S_dp[2][j][k] = dp[2][j][k] else: S_dp[2][j][k] = S_dp[2][j][k-1]+dp[2][j][k] S_dp[2][j][k] %= mod for i in range(3,N+1): for j in range(K): for k in range(K): if j < k: dp[i][j][k] = S_dp[i-1][k][k-1]-dp[i-1][k][j] elif j > k: dp[i][j][k] = S_dp[i-1][k][K-1]-S_dp[i-1][k][k]-dp[i-1][k][j] dp[i][j][k] %= mod if k == 0: S_dp[i][j][k] = dp[i][j][k] else: S_dp[i][j][k] = S_dp[i][j][k-1]+dp[i][j][k] S_dp[i][j][k] %= mod ans1 = 0 for j in range(K): ans1 += S_dp[-1][j][-1] ans1 %= mod ans2 = ans1*N*(K-1)*(mod+1)//2 ans2 %= mod print(ans1,ans2)