結果

問題 No.2351 Butterfly in Summer
コンテスト
ユーザー soldray
提出日時 2023-06-17 00:05:51
言語 C++17(clang)
(17.0.6 + boost 1.87.0)
結果
AC  
実行時間 2 ms / 2,000 ms
コード長 1,739 bytes
コンパイル時間 635 ms
コンパイル使用メモリ 119,808 KB
実行使用メモリ 5,376 KB
最終ジャッジ日時 2024-06-24 17:14:34
合計ジャッジ時間 1,461 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 21
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <stdio.h>
#include <iostream>
#include <vector>

using namespace std;

using i64 = long long;

#define yn(pope) (pope ? "Yes" : "No")

#define rep(i, n, m) for (int i = (n); i < (m); i++)
#define rrep(i, n, m) for (int i = (n); i > m; i--)

#define IINF (1 << 30)
#define INF (1ll << 60)

#define all(v) v.begin(), v.end()

i64 mod = 998244353ll;

struct mint {
  i64 x;

  mint(i64 x = 0):x(x % mod) {}

  mint& operator+=(const mint a) {
    x = (x + a.x) % mod;
    return *this;
  }

  mint& operator-=(const mint a) {
    x = (x - a.x + mod) % mod;
    return *this;
  }

  mint& operator*=(const mint a) {
    x = (x * a.x) % mod;
    return *this;
  }

  mint& operator/=(const mint a) {
    x = (x * a.inv().x) % mod;
    return *this;
  }

  mint operator+(const mint a) const {
    mint res(*this); res.x = (res.x + a.x) % mod;
    return res;
  }

  mint operator-(const mint a) const {
    mint res(*this); res.x = (res.x - a.x + mod) % mod;
    return res;
  }

  mint operator*(const mint a) const {
    mint res(*this); res.x = (res.x * a.x) % mod;
    return res;
  }

  mint operator/(const mint a) const {
    mint res(*this); res.x = (res.x * a.inv().x) % mod;
    return res;
  }

  mint pow(i64 pow) const {
    mint res = 1, base = x;

    while (pow) {
      if (pow & 1) {
	res *= base;
      }

      base *= base;
      pow >>= 1;
    }

    return res;
  }

  mint inv() const {
    return pow(mod - 2);
  }
};

istream& operator>>(istream& is, mint& a) { return is >> a.x;}
ostream& operator<<(ostream& os, const mint& a) { return os << a.x;}

int main() {
  int N, K; cin >> N >> K;

  mint ans = (mint)N * (mint)K * (mint)(K - 1);

  ans /= ((mint)K).pow(N);

  cout << ans.x << endl;

  return 0;
}
0