結果
問題 | No.2582 Random Average^K |
ユーザー | とりゐ |
提出日時 | 2023-11-03 23:43:58 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 326 ms / 2,000 ms |
コード長 | 1,110 bytes |
コンパイル時間 | 180 ms |
コンパイル使用メモリ | 81,700 KB |
実行使用メモリ | 91,200 KB |
最終ジャッジ日時 | 2023-12-09 23:30:36 |
合計ジャッジ時間 | 3,614 ms |
ジャッジサーバーID (参考情報) |
judge14 / judge15 |
(要ログイン)
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 62 ms
75,092 KB |
testcase_01 | AC | 62 ms
75,092 KB |
testcase_02 | AC | 64 ms
75,220 KB |
testcase_03 | AC | 62 ms
75,092 KB |
testcase_04 | AC | 66 ms
75,092 KB |
testcase_05 | AC | 62 ms
75,092 KB |
testcase_06 | AC | 86 ms
88,128 KB |
testcase_07 | AC | 77 ms
82,048 KB |
testcase_08 | AC | 85 ms
88,636 KB |
testcase_09 | AC | 153 ms
91,068 KB |
testcase_10 | AC | 191 ms
91,068 KB |
testcase_11 | AC | 175 ms
91,136 KB |
testcase_12 | AC | 319 ms
91,056 KB |
testcase_13 | AC | 296 ms
91,068 KB |
testcase_14 | AC | 61 ms
75,092 KB |
testcase_15 | AC | 326 ms
91,004 KB |
testcase_16 | AC | 204 ms
91,068 KB |
testcase_17 | AC | 206 ms
91,200 KB |
ソースコード
mod=998244353 table_size=10**6 fac=[1]*(table_size+1) finv=[1]*(table_size+1) for i in range(2,table_size+1): fac[i]=fac[i-1]*i%mod finv[table_size]=pow(fac[table_size],mod-2,mod) for i in range(table_size-1,-1,-1): finv[i]=finv[i+1]*(i+1)%mod def rebuild(n): global table_size,fac,finv fac+=[0]*(n-table_size) fac+=[0]*(n-table_size) finv+=[0]*(n-table_size) for i in range(table_size+1,n+1): fac[i]=fac[i-1]*i%mod finv[n]=inv(fac[n]) for i in range(n-1,table_size,-1): finv[i]=finv[i+1]*(i+1)%mod table_size=n def binom(n,k): if n<0 or k<0: return 0 if k>n: return 0 if n>table_size: rebuild(n+10**4) return (fac[n]*finv[k]%mod)*finv[n-k]%mod def fpow(x,k): res=1 while k: if k&1: res=res*x%mod x=x*x%mod k>>=1 return res def inv(a): if a<table_size: return fac[a-1]*finv[a]%mod return fpow(a,mod-2) N,K=map(int,input().split()) ans=0 if N%2: sgn=1 else: sgn=-1 for i in range(1,N+1): ans+=sgn*binom(N,i)*fpow(i,N+K) sgn=-sgn f=fpow(N,K) for i in range(K+1,K+N+1): f*=i f%=mod ans*=inv(f) print(ans%mod)