結果
問題 | No.3160 Party Game |
ユーザー |
|
提出日時 | 2025-05-23 22:01:52 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 1,711 ms / 2,000 ms |
コード長 | 1,234 bytes |
コンパイル時間 | 231 ms |
コンパイル使用メモリ | 82,404 KB |
実行使用メモリ | 121,404 KB |
最終ジャッジ日時 | 2025-05-27 22:05:21 |
合計ジャッジ時間 | 69,842 ms |
ジャッジサーバーID (参考情報) |
judge4 / judge3 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 38 |
ソースコード
import collections,sys,math,functools,operator,itertools,bisect,heapq,decimal,string,time,random input = sys.stdin.readline sys.set_int_max_str_digits(0) class comb(): def __init__(self,n,mod): self.fact = [0] * (n+1) self.fact_inv = [0] * (n+1) self.mod = mod self.fact[1] = 1 self.fact_inv[1] = pow(1,mod-2,mod) self.fact[0] = 1 self.fact_inv[0] = 1 for i in range(2,n+1): self.fact[i] = (self.fact[i-1] * i) % mod self.fact_inv[i] = (self.fact_inv[i-1] * pow(i,mod-2,mod)) % mod def ncr(self,n,r): if n-r < 0: return 0 return (self.fact[n] * self.fact_inv[r] * self.fact_inv[n-r]) % self.mod def nhr(self,n,r): return self.ncr(n+r-1,r) mod = 998244353 c = comb(2000000,mod) n,m = map(int,input().split()) if n == 1: ans = 0 minv = pow(m,mod-2,mod) for i in range(m): ans += i * minv ans %= mod print(ans) exit() ans = 0 p = 0 for i in range(m): x = m - i * n if x < 0:continue y = c.nhr(n+1,x) if i == 0: y -= n p = pow(y,mod-2,mod) continue #print(i,y,pow(p,mod-2,mod)) ans += p * y ans %= mod print(ans)