結果

問題 No.1100 Boxes
ユーザー yukudoyukudo
提出日時 2020-07-01 04:16:00
言語 C++14
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 30 ms / 2,000 ms
コード長 2,368 bytes
コンパイル時間 1,744 ms
コンパイル使用メモリ 165,908 KB
実行使用メモリ 8,680 KB
最終ジャッジ日時 2023-10-11 18:34:25
合計ジャッジ時間 5,038 ms
ジャッジサーバーID
（参考情報） 		judge15 / judge14
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 11 ms
8,436 KB
testcase_01 AC 11 ms
8,576 KB
testcase_02 AC 11 ms
8,448 KB
testcase_03 AC 11 ms
8,480 KB
testcase_04 AC 11 ms
8,388 KB
testcase_05 AC 11 ms
8,404 KB
testcase_06 AC 11 ms
8,356 KB
testcase_07 AC 10 ms
8,460 KB
testcase_08 AC 10 ms
8,572 KB
testcase_09 AC 11 ms
8,440 KB
testcase_10 AC 11 ms
8,360 KB
testcase_11 AC 11 ms
8,428 KB
testcase_12 AC 11 ms
8,336 KB
testcase_13 AC 11 ms
8,448 KB
testcase_14 AC 11 ms
8,508 KB
testcase_15 AC 10 ms
8,356 KB
testcase_16 AC 11 ms
8,424 KB
testcase_17 AC 11 ms
8,388 KB
testcase_18 AC 10 ms
8,348 KB
testcase_19 AC 11 ms
8,404 KB
testcase_20 AC 12 ms
8,576 KB
testcase_21 AC 16 ms
8,364 KB
testcase_22 AC 23 ms
8,324 KB
testcase_23 AC 16 ms
8,672 KB
testcase_24 AC 18 ms
8,332 KB
testcase_25 AC 20 ms
8,448 KB
testcase_26 AC 27 ms
8,644 KB
testcase_27 AC 23 ms
8,332 KB
testcase_28 AC 14 ms
8,428 KB
testcase_29 AC 25 ms
8,332 KB
testcase_30 AC 23 ms
8,576 KB
testcase_31 AC 17 ms
8,388 KB
testcase_32 AC 22 ms
8,448 KB
testcase_33 AC 30 ms
8,404 KB
testcase_34 AC 30 ms
8,508 KB
testcase_35 AC 11 ms
8,336 KB
testcase_36 AC 20 ms
8,428 KB
testcase_37 AC 11 ms
8,364 KB
testcase_38 AC 17 ms
8,680 KB
testcase_39 AC 28 ms
8,324 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

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

typedef long long ll;
#define REP(i,n) for(int i=0,_n=(int)(n);i<_n;++i)
#define ALL(v) (v).begin(),(v).end()
#define CLR(t,v) memset(t,(v),sizeof(t))
template<class T1,class T2>ostream& operator<<(ostream& os,const pair<T1,T2>&a){return os<<"("<<a.first<<","<<a.second<< ")";}
template<class T>void pv(T a,T b){for(T i=a;i!=b;++i)cout<<(*i)<<" ";cout<<endl;}
template<class T>void chmin(T&a,const T&b){if(a>b)a=b;}
template<class T>void chmax(T&a,const T&b){if(a<b)a=b;}

ll nextLong() { ll x; scanf("%lld", &x); return x;}


const ll MOD = 998244353;

const int SZ = 212345;
ll inv[SZ];
ll fact[SZ];
ll fact_inv[SZ];
ll choose(int n, int r) {
  if (n < 0 || r < 0 || n < r) return 0;
  return fact[n] * fact_inv[n-r] % MOD * fact_inv[r] % MOD;
}


struct mint {
  ll x;
  mint(ll x=0):x((x%MOD+MOD)%MOD){}
  mint& operator+=(const mint a) {if ((x += a.x) >= MOD) x -= MOD;return *this;}
  mint& operator-=(const mint a) {if ((x += MOD-a.x) >= MOD) x -= MOD;return *this;}
  mint& operator*=(const mint a) {(x *= a.x) %= MOD;return *this;}
  mint operator+(const mint a) const {mint res(*this);return res+=a;}
  mint operator-(const mint a) const {mint res(*this);return res-=a;}
  mint operator*(const mint a) const {mint res(*this);return res*=a;}
  mint pow(ll b) const {
    mint res(1), a(*this);
    while (b) {
      if (b & 1) res *= a;
      a *= a;
      b >>= 1;
    }
    return res;
  }
  // for prime MOD
  mint inv() const {return pow(MOD-2);}
  mint& operator/=(const mint a) {return (*this) *= a.inv();}
  mint operator/(const mint a) const {mint res(*this);return res/=a;}
};
ostream& operator<<(ostream& os, const mint& a) {os << a.x; return os;}


int main2() {
  ll N = nextLong();
  ll K = nextLong();

  mint ans = 0;
  for (int k = 1; k <= K; k++) {
    mint val = mint(choose(K, k)) * mint(2).pow(k-1);
    val *= mint(K - k).pow(N);
    if (k % 2)
      ans += val;
    else
      ans -= val;
  }
  cout << ans << endl;
  return 0;
}

int main() {
  inv[1] = 1;
  for (int i = 2; i < SZ; i++) inv[i] = inv[(int) (MOD % i)] * (MOD - MOD / i) % MOD;
  fact[0] = 1;
  for (int i = 1; i < SZ; i++) fact[i] = fact[i-1] * i % MOD;
  fact_inv[0] = 1;
  for (int i = 1; i < SZ; i++) fact_inv[i] = fact_inv[i-1] * inv[i] % MOD;

#ifdef LOCAL
  for (;!cin.eof();cin>>ws)
#endif
    main2();
  return 0;
}
0