結果
問題 | No.2582 Random Average^K |
ユーザー |
|
提出日時 | 2023-12-10 00:10:02 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 1,540 ms / 2,000 ms |
コード長 | 1,130 bytes |
コンパイル時間 | 261 ms |
コンパイル使用メモリ | 82,312 KB |
実行使用メモリ | 321,980 KB |
最終ジャッジ日時 | 2025-02-21 06:13:00 |
合計ジャッジ時間 | 22,422 ms |
ジャッジサーバーID (参考情報) |
judge4 / judge1 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 15 |
ソースコード
import sysfrom itertools import permutationsfrom heapq import heappop,heappushfrom collections import dequeimport randomimport bisectinput = lambda :sys.stdin.readline().rstrip()mi = lambda :map(int,input().split())li = lambda :list(mi())"""E[prod x_i^a_i] = prod 1/(ai+1)sum a_i = K を満たすすべての a に対するK! * prod 1/(ai+1)! の総和1/1!+x/2!+x^2/3!+... = (e^x-1)/xK! [x^K] (e^x-1)^N/x^N= K! [x^(N+K)] (e^x-1)^N"""def cmb(n, r, mod):if ( r<0 or r>n ):return 0return (g1[n] * g2[r] % mod) * g2[n-r] % modmod = 998244353N = 10**6 + 100 + 10**7g1 = [1]*(N+1)g2 = [1]*(N+1)inverse = [1]*(N+1)for i in range( 2, N + 1 ):g1[i]=( ( g1[i-1] * i ) % mod )inverse[i]=( ( -inverse[mod % i] * (mod//i) ) % mod )g2[i]=( (g2[i-1] * inverse[i]) % mod )inverse[0]=0N,K = mi()res = 0for i in range(N+1):if (N-i) & 1:res -= cmb(N,i,mod) * pow(i,N+K,mod) % modelse:res += cmb(N,i,mod) * pow(i,N+K,mod) % modres %= modres *= g1[K] * g2[N+K] % modres %= modres *= pow(inverse[N],K,mod)res %= modprint(res % mod)