結果
問題 |
No.2245 Second Smallest
|
ユーザー |
|
提出日時 | 2023-03-10 23:24:04 |
言語 | C++14 (gcc 13.3.0 + boost 1.87.0) |
結果 |
TLE
|
実行時間 | - |
コード長 | 1,655 bytes |
コンパイル時間 | 2,417 ms |
コンパイル使用メモリ | 113,484 KB |
実行使用メモリ | 10,624 KB |
最終ジャッジ日時 | 2024-09-18 05:39:22 |
合計ジャッジ時間 | 6,035 ms |
ジャッジサーバーID (参考情報) |
judge2 / judge5 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
other | AC * 5 TLE * 1 -- * 24 |
ソースコード
#include<iostream> #include<atcoder/modint> #include<atcoder/convolution> using namespace std; #include<vector> template<typename T> struct combination{ vector<T>fac,ifac; combination(size_t N=0):fac(1,1),ifac(1,1) { make_table(N); } void make_table(size_t N) { if(fac.size()>N)return; size_t now=fac.size(); N=max(N,now*2); fac.resize(N+1); ifac.resize(N+1); for(size_t i=now;i<=N;i++)fac[i]=fac[i-1]*i; ifac[N]=1/fac[N]; for(size_t i=N;i-->now;)ifac[i]=ifac[i+1]*(i+1); } T factorial(size_t n) { make_table(n); return fac[n]; } T invfac(size_t n) { make_table(n); return ifac[n]; } T P(size_t n,size_t k) { if(n<k)return 0; make_table(n); return fac[n]*ifac[n-k]; } T C(size_t n,size_t k) { if(n<k)return 0; make_table(n); return fac[n]*ifac[n-k]*ifac[k]; } T H(size_t n,size_t k) { if(n==0)return k==0?1:0; return C(n-1+k,k); } }; using mint=atcoder::modint998244353; combination<mint>C; int N,M,K; int main() { cin>>N>>M>>K; int T=min(N,M); vector<mint>dp(T+1); for(int c=0;c<=T;c++) { const mint coef=C.C(N,c)*C.C(M,c); //x^c * (1-x)^(T-c) for(int i=0;i<=T-c;i++) { if(i%2==0)dp[i+c]-=C.C(T-c,i)*coef; else dp[i+c]+=C.C(T-c,i)*coef; } } vector<mint>E(1,mint::raw(1)),A(2); A[0]=mint::raw(1);A[1]=-1; { int K=N*M-T; while(K) { if(K&1) { E=atcoder::convolution(E,A); } K>>=1; if(K) { A=atcoder::convolution(A,A); } } } dp=atcoder::convolution(dp,E); for(int i=0;i+1<dp.size();i++)dp[i]=dp[i+1]*mint::raw(i+1); dp.pop_back(); mint ans=0; for(int i=0;i<dp.size();i++) { ans+=dp[i]/(i+K+1); } cout<<ans.val()<<endl; }