Problem p02685 (C++)

Buggy submission (status: Runtime Error):
#include <bits/stdc++.h>
#define rep(i, n) for (int i = 0; i < (n); i++)
#define reps(i, s, n) for (int i = (s); i < (n); i++)
#define all(v) v.begin(), v.end()
#define outve(v)                                                               \
  for (auto i : v)                                                             \
    cout << i << " ";                                                          \
  cout << endl
#define outmat(v)                                                              \
  for (auto i : v) {                                                           \
    for (auto j : i)                                                           \
      cout << j << " ";                                                        \
    cout << endl;                                                              \
  }
#define in(n, v)                                                               \
  for (int i = 0; i < (n); i++) {                                              \
    cin >> v[i];                                                               \
  }
#define out(n) cout << (n) << endl
#define fi first
#define se second
#define pb push_back
#define si(v) int(v.size())
#define len(v) int(v.length())
#define lob(v, n) lower_bound(all(v), n)
#define lobi(v, n) lower_bound(all(v), n) - v.begin()
#define upb(v, n) upper_bound(all(v), n)
#define upbi(v, n) upper_bound(all(v), n) - v.begin()
#define mod 998244353
#define infi 1900000000
#define infl 1100000000000000000
#define cyes cout << "Yes" << endl
#define cno cout << "No" << endl
#define csp << " " <<
#define outset(n) cout << fixed << setprecision(n);
using namespace std;
using ll = long long;
using ld = long double;
using vi = vector<int>;
using vvi = vector<vector<int>>;
using vs = vector<string>;
using vc = vector<char>;
using vd = vector<double>;
using vvd = vector<vector<double>>;
using vl = vector<ll>;
using vvl = vector<vector<ll>>;
using pii = pair<int, int>;
using pll = pair<ll, ll>;
using vvp = vector<vector<pair<int, int>>>;
template <typename T> using ve = vector<T>;
template <typename T> using pq2 = priority_queue<T>;
template <typename T> using pq1 = priority_queue<T, vector<T>, greater<T>>;
template <typename T> bool chmax(T &a, T b) {
  if (a < b) {
    a = b;
    return 1;
  }
  return 0;
}
template <typename T> bool chmin(T &a, T b) {
  if (a > b) {
    a = b;
    return 1;
  }
  return 0;
}

template <typename T> T gcd(T a, T b) {
  if (b == 0)
    return a;
  return gcd(b, a % b);
}
template <typename T> T lcm(T a, T b) { return a * b / gcd(a, b); }
template <typename T> T extgcd(T a, T b, T &x, T &y) {
  T d = a;
  if (b != 0) {
    d = extgcd(b, a % b, y, x);
    y -= (a / b) * x;
  } else
    x = 1, y = 0;
  return d;
}
template <typename T> T modpow(T x, T n, T m) {
  if (n == 0)
    return 1;
  T resp = modpow(x * x % m, n / 2, m);
  if (n & 1)
    resp = resp * x % m;
  return resp;
}
template <typename T> T modinverse(T a, T m) {
  T x, y;
  extgcd(a, m, x, y);
  return (m + x % m) % m;
}
struct combination {
  ll M = mod;
  vector<ll> fact, ifact;
  combination(ll n) : fact(n + 1), ifact(n + 1) {
    fact[0] = 1;
    ifact[0] = 1;
    for (ll i = 0; i < n; i++) {
      fact[i + 1] = fact[i] * (i + 1) % M;
      ifact[i + 1] = ifact[i] * modpow(i + 1, M - 2, M) % M;
    }
  }
  ll com(ll a, ll b) {
    if (a == 0 && b == 0)
      return 1;
    if (a < b || a < 0)
      return 0;
    ll tmp = ifact[a - b] * ifact[b] % M;
    return tmp * fact[a] % M;
  }
};

int main() {

  ios::sync_with_stdio(false);
  cin.tie(nullptr);

  ll N, M, K;
  cin >> N >> M >> K;

  ll ans = 0;

  combination C(M + 1);

  for (ll k = 0; k <= K; k++) {
    ll kk = modpow<ll>(M - 1, N - 1 - k, mod);
    ll kkk = C.com(N - 1, k);
    ans += kk * kkk % mod;
    ans %= mod;
  }
  ans *= M;
  ans %= mod;

  cout << ans << endl;

  return 0;
}


Fixed submission:
#include <bits/stdc++.h>
#define rep(i, n) for (int i = 0; i < (n); i++)
#define reps(i, s, n) for (int i = (s); i < (n); i++)
#define all(v) v.begin(), v.end()
#define outve(v)                                                               \
  for (auto i : v)                                                             \
    cout << i << " ";                                                          \
  cout << endl
#define outmat(v)                                                              \
  for (auto i : v) {                                                           \
    for (auto j : i)                                                           \
      cout << j << " ";                                                        \
    cout << endl;                                                              \
  }
#define in(n, v)                                                               \
  for (int i = 0; i < (n); i++) {                                              \
    cin >> v[i];                                                               \
  }
#define out(n) cout << (n) << endl
#define fi first
#define se second
#define pb push_back
#define si(v) int(v.size())
#define len(v) int(v.length())
#define lob(v, n) lower_bound(all(v), n)
#define lobi(v, n) lower_bound(all(v), n) - v.begin()
#define upb(v, n) upper_bound(all(v), n)
#define upbi(v, n) upper_bound(all(v), n) - v.begin()
#define mod 998244353
#define infi 1900000000
#define infl 1100000000000000000
#define cyes cout << "Yes" << endl
#define cno cout << "No" << endl
#define csp << " " <<
#define outset(n) cout << fixed << setprecision(n);
using namespace std;
using ll = long long;
using ld = long double;
using vi = vector<int>;
using vvi = vector<vector<int>>;
using vs = vector<string>;
using vc = vector<char>;
using vd = vector<double>;
using vvd = vector<vector<double>>;
using vl = vector<ll>;
using vvl = vector<vector<ll>>;
using pii = pair<int, int>;
using pll = pair<ll, ll>;
using vvp = vector<vector<pair<int, int>>>;
template <typename T> using ve = vector<T>;
template <typename T> using pq2 = priority_queue<T>;
template <typename T> using pq1 = priority_queue<T, vector<T>, greater<T>>;
template <typename T> bool chmax(T &a, T b) {
  if (a < b) {
    a = b;
    return 1;
  }
  return 0;
}
template <typename T> bool chmin(T &a, T b) {
  if (a > b) {
    a = b;
    return 1;
  }
  return 0;
}

template <typename T> T gcd(T a, T b) {
  if (b == 0)
    return a;
  return gcd(b, a % b);
}
template <typename T> T lcm(T a, T b) { return a * b / gcd(a, b); }
template <typename T> T extgcd(T a, T b, T &x, T &y) {
  T d = a;
  if (b != 0) {
    d = extgcd(b, a % b, y, x);
    y -= (a / b) * x;
  } else
    x = 1, y = 0;
  return d;
}
template <typename T> T modpow(T x, T n, T m) {
  if (n == 0)
    return 1;
  T resp = modpow(x * x % m, n / 2, m);
  if (n & 1)
    resp = resp * x % m;
  return resp;
}
template <typename T> T modinverse(T a, T m) {
  T x, y;
  extgcd(a, m, x, y);
  return (m + x % m) % m;
}
struct combination {
  ll M = mod;
  vector<ll> fact, ifact;
  combination(ll n) : fact(n + 1), ifact(n + 1) {
    fact[0] = 1;
    ifact[0] = 1;
    for (ll i = 0; i < n; i++) {
      fact[i + 1] = fact[i] * (i + 1) % M;
      ifact[i + 1] = ifact[i] * modpow(i + 1, M - 2, M) % M;
    }
  }
  ll com(ll a, ll b) {
    if (a == 0 && b == 0)
      return 1;
    if (a < b || a < 0)
      return 0;
    ll tmp = ifact[a - b] * ifact[b] % M;
    return tmp * fact[a] % M;
  }
};

int main() {

  ios::sync_with_stdio(false);
  cin.tie(nullptr);

  ll N, M, K;
  cin >> N >> M >> K;

  ll ans = 0;

  combination C(N + 1);

  for (ll k = 0; k <= K; k++) {
    ll kk = modpow<ll>(M - 1, N - 1 - k, mod);
    ll kkk = C.com(N - 1, k);
    ans += kk * kkk % mod;
    ans %= mod;
  }
  ans *= M;
  ans %= mod;

  cout << ans << endl;

  return 0;
}