結果
問題 |
No.3174 勝ち残りじゃんけん
|
ユーザー |
|
提出日時 | 2025-06-06 23:13:36 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 1,290 ms / 2,000 ms |
コード長 | 1,816 bytes |
コンパイル時間 | 391 ms |
コンパイル使用メモリ | 82,688 KB |
実行使用メモリ | 273,920 KB |
最終ジャッジ日時 | 2025-06-06 23:13:47 |
合計ジャッジ時間 | 9,577 ms |
ジャッジサーバーID (参考情報) |
judge1 / judge3 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 17 |
ソースコード
import sys input = lambda :sys.stdin.readline()[:-1] ni = lambda :int(input()) na = lambda :list(map(int,input().split())) yes = lambda :print("yes");Yes = lambda :print("Yes");YES = lambda : print("YES") no = lambda :print("no");No = lambda :print("No");NO = lambda : print("NO") ####################################################################### mod = 998244353 nn = 5010 fact = [1] * nn for i in range(nn - 1): fact[i + 1] = fact[i] * (i + 1) % mod invfact = [1] * nn invfact[nn - 1] = pow(fact[nn - 1], mod - 2, mod) for i in range(nn - 1)[::-1]: invfact[i] = invfact[i + 1] * (i + 1) % mod def binom(x, y): if x < 0 or y < 0 or x - y < 0: return 0 return fact[x] * invfact[y] % mod * invfact[x - y] % mod n = ni() m3 = pow(3, mod-2, mod) pi3 = [1] for i in range(n): pi3.append(pi3[-1] * m3 % mod) dpp = [0] * (n + 1) dpe = [0] * (n + 1) f = [0] * (n + 1) for i in range(1, n + 1): for j in range(1, i): f[i] += binom(i, j) * pi3[i-1] % mod f[i] %= mod f[i] = pow(f[i], mod-2, mod) # print([f[i] for i in range(n+1)]) dpp[n] = 1 ans = [[0] * ( 1 + n) for i in range(n+1)] rui = [0] * (n + 1) for i in range(n, 1, -1): for x in range(1, i): y = binom(i, x) * pi3[i-1] % mod * f[i] % mod # print(i, x, y ) dpp[i - x] += dpp[i] * y % mod dpp[i-x] %= mod dpe[i - x] += dpe[i] * y % mod + f[i] * dpp[i] % mod * y % mod dpe[i-x] %= mod ans[i][i-x] += dpe[i] * y % mod + f[i] * dpp[i] % mod * y % mod ans[i][i-x] %= mod # print(dpp[1:]) # print(dpe[1:]) res = 0 g = [] for i in range(1, n + 1): for j in range(i): res -= ans[i][j] res %= mod for j in range(i+1, n+1): res += ans[j][i] res %= mod g.append(res) print(*g)