結果

問題 No.2130 分配方法の数え上げ mod 998244353
コンテスト
ユーザー Nzt3
提出日時 2022-11-25 22:46:34
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 4 ms / 2,000 ms
コード長 759 bytes
コンパイル時間 1,610 ms
コンパイル使用メモリ 193,448 KB
最終ジャッジ日時 2025-02-09 00:35:49
ジャッジサーバーID
（参考情報） 		judge3 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 38
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include<bits/stdc++.h>
using namespace std;
const int mod=998244353;

long long fac[100005];
long long inv[100005];
long long invf[100005];
long long modpow(long long a,long long n){
  long long ret=1,t=a;
  n%=mod-1;
  for(int i=0;i<30;i++){
    if(n>>i&1)ret=ret*t%mod;
    t=t*t%mod;
  }
  return ret;
}

int main(){
  ios::sync_with_stdio(false);
  cin.tie(nullptr);
  long long N,M;
  cin>>N>>M;
  fac[0]=inv[0]=invf[0]=1;
  fac[1]=inv[1]=invf[1]=1;
  for(int i=2;i<100005;i++){
    fac[i]=fac[i-1]*i%mod;
    inv[i]=inv[mod%i]*(mod-mod/i)%mod;
    invf[i]=invf[i-1]*inv[i]%mod;
  }
  long long ans=modpow(2,N),k=1;
  for(int i=0;i<min(N+1,M);i++){
    ans=(ans-k)%mod;
    k=k*((N-i)%mod)%mod*inv[i+1]%mod;
  }
  if(ans<0)ans+=mod;
  cout<<ans<<'\n';
}
0