結果

問題 No.1100 Boxes
ユーザー cureskolcureskol
提出日時 2022-07-15 21:28:30
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 113 ms / 2,000 ms
コード長 860 bytes
コンパイル時間 2,283 ms
コンパイル使用メモリ 195,932 KB
最終ジャッジ日時 2025-01-30 07:28:05
ジャッジサーバーID
（参考情報） 		judge5 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 4
other AC * 36
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
using namespace std;

#define REP(i,n) for(int i=0;i<(n);i++)

#include <atcoder/modint>
using namespace atcoder;
using mint=modint998244353;
ostream& operator<<(ostream &os,mint a){os<<a.val();return os;}
istream& operator>>(istream &is,mint &a){
  long long b;is>>b;a=b;
  return is;
}

const int SIZE=5000000;
mint Factorial[SIZE],Finverse[SIZE];
void Cinit(){
  Factorial[0]=mint(1);
  for(int i=1;i<SIZE;i++)Factorial[i]=Factorial[i-1]*mint(i);
  Finverse[SIZE-1]=Factorial[SIZE-1].inv();
  for(int i=SIZE-2;i>=0;i--)Finverse[i]=mint(i+1)*Finverse[i+1];
}
mint nCk(int n,int k){
  if(Factorial[0]==0)Cinit();
  if(n<k)return 0;
  return Factorial[n]*Finverse[n-k]*Finverse[k];
}

int main(){
  int n,k;cin>>n>>k;
  mint ans=mint(k).pow(n);
  REP(i,k+1)ans-=mint(k-i).pow(n)*mint(-2).pow(i)*nCk(k,i);
  ans/=2;
  cout<<ans<<endl;
}
0