結果

問題 No.1856 Mex Sum 2
ユーザー ytqm3ytqm3
提出日時 2022-02-25 23:07:24
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 445 ms / 3,000 ms
コード長 1,006 bytes
コンパイル時間 5,108 ms
コンパイル使用メモリ 258,204 KB
最終ジャッジ日時 2025-01-28 02:33:00
ジャッジサーバーID
（参考情報） 		judge4 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 64
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include<bits/stdc++.h>
#include<atcoder/all>
typedef uint64_t u64;
typedef int64_t i64;
template<u64 mod> using modint=atcoder::static_modint<mod>;
using namespace std;

template<typename T> struct comb{
  vector<T> dat,idat;
  comb(int mx=3000000):dat(mx+1,1),idat(mx+1,1){
    for(int i=1;i<=mx;++i){
      dat[i]=dat[i-1]*i;
    }
    idat[mx]/=dat[mx];
    for(int i=mx;i>0;--i){
      idat[i-1]=idat[i]*i;
    }
  }
  T operator()(int n,int k){
    if(n<0||k<0||n<k){
      return 0;
    }
    return dat[n]*idat[k]*idat[n-k];
  }
};

int main(){
  constexpr u64 mod=998244353;
  typedef modint<mod> mint;
  comb<mint> C;
  int N,M;
  cin>>N>>M;
  vector<mint> f(N+1);
  for(int i=0;i<=N;++i){
    f[i]=(mint(2).pow(i)-1)*C.idat[i];
  }
  vector<mint> g(N+1);
  g[0]=1;
  mint ans=0;
  for(int i=0;i<min(N,M+1);++i){
  	g=atcoder::convolution(f,g);
  	g.resize(N+1);
  	mint now=1;
  	for(int j=N;j>i;--j){
  		ans+=C.dat[N]*C.idat[N-j]*now*g[j];
  		now*=2*(M-i);
  	}
  }
  cout<<ans.val()<<endl;
}
0