結果
問題 | No.1392 Don't be together |
ユーザー | convexineq |
提出日時 | 2021-02-12 22:50:36 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 398 ms / 2,000 ms |
コード長 | 1,205 bytes |
コンパイル時間 | 1,166 ms |
コンパイル使用メモリ | 87,068 KB |
実行使用メモリ | 79,940 KB |
最終ジャッジ日時 | 2023-09-27 05:48:25 |
合計ジャッジ時間 | 8,751 ms |
ジャッジサーバーID (参考情報) |
judge13 / judge12 |
(要ログイン)
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 71 ms
71,560 KB |
testcase_01 | AC | 69 ms
71,552 KB |
testcase_02 | AC | 71 ms
71,320 KB |
testcase_03 | AC | 71 ms
71,316 KB |
testcase_04 | AC | 69 ms
71,384 KB |
testcase_05 | AC | 70 ms
71,536 KB |
testcase_06 | AC | 193 ms
77,896 KB |
testcase_07 | AC | 393 ms
79,212 KB |
testcase_08 | AC | 334 ms
79,108 KB |
testcase_09 | AC | 398 ms
79,940 KB |
testcase_10 | AC | 280 ms
77,840 KB |
testcase_11 | AC | 380 ms
78,916 KB |
testcase_12 | AC | 266 ms
77,808 KB |
testcase_13 | AC | 333 ms
78,476 KB |
testcase_14 | AC | 359 ms
78,664 KB |
testcase_15 | AC | 275 ms
77,852 KB |
testcase_16 | AC | 386 ms
78,996 KB |
testcase_17 | AC | 287 ms
78,020 KB |
testcase_18 | AC | 357 ms
78,624 KB |
testcase_19 | AC | 324 ms
78,376 KB |
testcase_20 | AC | 197 ms
78,096 KB |
testcase_21 | AC | 133 ms
77,464 KB |
testcase_22 | AC | 255 ms
77,928 KB |
testcase_23 | AC | 205 ms
77,868 KB |
testcase_24 | AC | 159 ms
77,564 KB |
testcase_25 | AC | 213 ms
77,888 KB |
testcase_26 | AC | 128 ms
77,944 KB |
testcase_27 | AC | 143 ms
77,448 KB |
testcase_28 | AC | 224 ms
77,760 KB |
testcase_29 | AC | 162 ms
77,548 KB |
ソースコード
MOD = 998244353 n,m = map(int,input().split()) *p, = map(int,input().split()) for i in range(n): p[i] -= 1 #from collections import Counter #d = Counter() res = [] for i in range(n): if p[i] == -1: continue c = 0 while p[i] != -1: ni = p[i] p[i] = -1 i = ni c += 1 res.append(c) #d[c] += 1 dp = [0]*(m+1) dp[0] = 1 for c in res: #print(c,dp) ndp = [0]*(m+1) for i in range(m+1): ndp[i] += i*dp[i] ndp[i] %= MOD if i < m: ndp[i+1] += dp[i] dp = ndp #print(dp,"now") dp2 = [0]*(m+1) for _ in range(c-2): ndp = [0]*(m+1) ndp2 = [0]*(m+1) for i in range(1,m+1): ndp2[i] += dp[i]*(i-1) ndp2[i] += dp2[i]*(i-2) ndp2[i] %= MOD ndp[i] += dp2[i] ndp[i] %= MOD if i < m: ndp2[i+1] += dp[i] + dp2[i] dp = ndp dp2 = ndp2 #print(dp,dp2) ndp = [0]*(m+1) for i in range(m+1): ndp[i] += (i-1)*dp[i] ndp[i] += (i-2)*dp2[i] ndp[i] %= MOD if i < m: ndp[i+1] += dp[i] + dp2[i] ndp[i+1] %= MOD dp = ndp print(dp[-1])