結果
問題 | No.2582 Random Average^K |
ユーザー | umezo |
提出日時 | 2023-12-11 02:59:11 |
言語 | C++17 (gcc 12.3.0 + boost 1.83.0) |
結果 |
AC
|
実行時間 | 946 ms / 2,000 ms |
コード長 | 1,077 bytes |
コンパイル時間 | 2,135 ms |
コンパイル使用メモリ | 202,928 KB |
実行使用メモリ | 472,308 KB |
最終ジャッジ日時 | 2023-12-11 02:59:35 |
合計ジャッジ時間 | 19,011 ms |
ジャッジサーバーID (参考情報) |
judge15 / judge14 |
(要ログイン)
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 915 ms
472,308 KB |
testcase_01 | AC | 946 ms
472,308 KB |
testcase_02 | AC | 760 ms
472,308 KB |
testcase_03 | AC | 787 ms
472,308 KB |
testcase_04 | AC | 767 ms
472,308 KB |
testcase_05 | AC | 795 ms
472,308 KB |
testcase_06 | AC | 793 ms
472,308 KB |
testcase_07 | AC | 762 ms
472,308 KB |
testcase_08 | AC | 793 ms
472,308 KB |
testcase_09 | AC | 813 ms
472,308 KB |
testcase_10 | AC | 800 ms
472,308 KB |
testcase_11 | AC | 823 ms
472,308 KB |
testcase_12 | AC | 892 ms
472,308 KB |
testcase_13 | AC | 847 ms
472,308 KB |
testcase_14 | AC | 777 ms
472,308 KB |
testcase_15 | AC | 881 ms
472,308 KB |
testcase_16 | AC | 805 ms
472,308 KB |
testcase_17 | AC | 821 ms
472,308 KB |
ソースコード
#define rep(i,n) for(int i=0;i<(int)(n);i++) #define ALL(v) v.begin(),v.end() typedef long long ll; #include<bits/stdc++.h> using namespace std; const int MAX=20002000; const int MOD=998244353; ll fac[MAX],finv[MAX],inv[MAX]; void init(){ fac[0]=fac[1]=1; finv[0]=finv[1]=1; inv[1]=1; for(int i=2;i<MAX;i++){ fac[i]=fac[i-1]*i%MOD; inv[i]=MOD-inv[MOD%i]*(MOD/i)%MOD; finv[i]=finv[i-1]*inv[i]%MOD; } } ll nCr(int n,int k){ if(n<k) return 0; if(n<0 || k<0) return 0; return fac[n]*(finv[k]*finv[n-k]%MOD)%MOD; } ll modpow(ll x,ll n){ ll ans=1; while(n){ if(n&1) ans=ans*x%MOD; x=x*x%MOD; n/=2; } return ans; } ll nSr(int n,int k){ ll tmp=0; for(int i=0;i<=k;i++){ if((k-i)%2==0) tmp=(tmp+nCr(k,i)*modpow(i,n)%MOD)%MOD; else tmp=(tmp-nCr(k,i)*modpow(i,n)%MOD+MOD)%MOD; } return tmp*finv[k]%MOD; } int main(){ ios::sync_with_stdio(false); std::cin.tie(nullptr); init (); ll n,k; cin>>n>>k; ll tmp=modpow(n,k)*nCr(n+k,k)%MOD; cout<<nSr(n+k,n)*modpow(tmp,MOD-2)%MOD<<endl; return 0; }