Problem p02685 (C++) — buggy submission, verdict: Time Limit Exceeded

#include <algorithm> // min, max, swap, sort, reverse, lower_bound, upper_bound
#include <bitset>    // bitset
#include <cctype>    // isupper, islower, isdigit, toupper, tolower
#include <cmath>
#include <cstdint>  // int64_t, int*_t
#include <cstdio>   // printf
#include <deque>    // deque
#include <iostream> // cout, endl, cin
#include <map>      // map
#include <math.h>
#include <queue>         // queue, priority_queue
#include <set>           // set
#include <stack>         // stack
#include <string>        // string, to_string, stoi
#include <tuple>         // tuple, make_tuple
#include <unordered_map> // unordered_map
#include <unordered_set> // unordered_set
#include <utility>       // pair, make_pair
#include <vector>        // vector
using namespace std;

#define int long long
#define endl "\\n"
#define all(v) v.begin(), v.end()
#define fir first
#define sec second
#define m_p make_pair
#define rep(i, n) for (int i = 0; i < (int)(n); i++)
#define vecin(v)                                                               \\
  for (int i = 0; i < (int)v.size(); i++)                                      \\
    cin >> v[i];
#define vecout(v, n)                                                           \\
  for (int i = 0; i < n; i++)                                                  \\
    cout << v[i] << " ";                                                       \\
  cout << endl;

const int mod = 998244353;
const int INF = 2000000015;
const int MAX = 510000;
const int MOD = 998244353;
long long fac[MAX], finv[MAX], inv[MAX];

// テーブルを作る前処理
void COMinit() {
  fac[0] = fac[1] = 1;
  finv[0] = finv[1] = 1;
  inv[1] = 1;
  for (int i = 2; i < MAX; i++) {
    fac[i] = fac[i - 1] * i % MOD;
    inv[i] = MOD - inv[MOD % i] * (MOD / i) % MOD;
    finv[i] = finv[i - 1] * inv[i] % MOD;
  }
}

// 二項係数計算
long long COM(int n, int k) {
  if (n < k)
    return 0;
  if (n < 0 || k < 0)
    return 0;
  return fac[n] * (finv[k] * finv[n - k] % MOD) % MOD;
}
int jousu(int x00, int y00) {
  int z00 = 1;
  for (int i = 0; i < y00; i++) {
    z00 *= x00;
  }
  return z00;
}
int keta(int x00) {
  int z00 = x00;
  int w00 = 0;
  while (z00 != 0) {
    z00 /= 10;
    w00++;
  }
  return w00;
}
int krag(double xx) { return INF - (int)(INF - xx); }

// x未満の要素数を返す二分探索関数
int b_s(vector<int> &vecxx, int xx) {
  if (xx <= vecxx[0]) { // x以下の要素数を返させる時は"<"に変える
    return 0;
  }
  if (vecxx.back() < xx) { // x以下の要素数を返させる時は"<="に変える
    return vecxx.size();
  }
  int aa1 = 0, aa2 = vecxx.size() - 1, aa3;
  while (aa1 + 1 < aa2) {
    aa3 = (aa1 + aa2) / 2;
    if (vecxx[aa3] >= xx) { // x以下の要素数を返させる時は">"に変える
      aa2 = aa3;
    } else {
      aa1 = aa3;
    }
  }
  return aa2;
}

int modinv(int a, int m) {
  int b = m, u = 1, v = 0;
  while (b) {
    int t = a / b;
    a -= t * b;
    swap(a, b);
    u -= t * v;
    swap(u, v);
  }
  u %= m;
  if (u < 0)
    u += m;
  return u;
}
int modpow(int a, int n) {
  int res = 1;
  while (n > 0) {
    if (n & 1)
      res = res * a % mod;
    a = a * a % mod;
    n >>= 1;
  }
  return res;
}
// n個を1個以上のx組のグループに分ける重複組み合わせはcom(n-1,x-1)
// グループが0個でもいいときはnにxを足す

int gcd(int xx, int yy) {
  int p = xx;
  int q = yy;
  if (q > p)
    swap(p, q);
  while (p % q != 0) {
    p %= q;
    swap(p, q);
  }
  return q;
}
int lcm(int xx, int yy) { return xx * yy / gcd(xx, yy); }
bool prime(int xx) {
  int a = xx;
  for (int i = 2; i * i <= xx; i++) {
    if (xx % i == 0) {
      return 0;
    }
  }
  return 1;
}

signed main() {
  COMinit();
  int n, m, k, ans = 0;
  cin >> n >> m >> k;
  for (int i = 0; i <= k; i++) {
    ans += ((COM(n - 1, i) * m) % mod) * modpow(m - 1, n - i - 1);
    ans %= mod;
  }
  cout << ans << endl;
}


Error: TLE

Fixed submission:

#include <algorithm> // min, max, swap, sort, reverse, lower_bound, upper_bound
#include <bitset>    // bitset
#include <cctype>    // isupper, islower, isdigit, toupper, tolower
#include <cmath>
#include <cstdint>  // int64_t, int*_t
#include <cstdio>   // printf
#include <deque>    // deque
#include <iostream> // cout, endl, cin
#include <map>      // map
#include <math.h>
#include <queue>         // queue, priority_queue
#include <set>           // set
#include <stack>         // stack
#include <string>        // string, to_string, stoi
#include <tuple>         // tuple, make_tuple
#include <unordered_map> // unordered_map
#include <unordered_set> // unordered_set
#include <utility>       // pair, make_pair
#include <vector>        // vector
using namespace std;

#define int long long
#define endl "\\n"
#define all(v) v.begin(), v.end()
#define fir first
#define sec second
#define m_p make_pair
#define rep(i, n) for (int i = 0; i < (int)(n); i++)
#define vecin(v)                                                               \\
  for (int i = 0; i < (int)v.size(); i++)                                      \\
    cin >> v[i];
#define vecout(v, n)                                                           \\
  for (int i = 0; i < n; i++)                                                  \\
    cout << v[i] << " ";                                                       \\
  cout << endl;

const int mod = 998244353;
const int INF = 2000000015;
const int MAX = 510000;
const int MOD = 1000000007;
long long fac[MAX], finv[MAX], inv[MAX];
int jousu(int x00, int y00) {
  int z00 = 1;
  for (int i = 0; i < y00; i++) {
    z00 *= x00;
  }
  return z00;
}
int keta(int x00) {
  int z00 = x00;
  int w00 = 0;
  while (z00 != 0) {
    z00 /= 10;
    w00++;
  }
  return w00;
}
int krag(double xx) { return INF - (int)(INF - xx); }

// x未満の要素数を返す二分探索関数
int b_s(vector<int> &vecxx, int xx) {
  if (xx <= vecxx[0]) { // x以下の要素数を返させる時は"<"に変える
    return 0;
  }
  if (vecxx.back() < xx) { // x以下の要素数を返させる時は"<="に変える
    return vecxx.size();
  }
  int aa1 = 0, aa2 = vecxx.size() - 1, aa3;
  while (aa1 + 1 < aa2) {
    aa3 = (aa1 + aa2) / 2;
    if (vecxx[aa3] >= xx) { // x以下の要素数を返させる時は">"に変える
      aa2 = aa3;
    } else {
      aa1 = aa3;
    }
  }
  return aa2;
}

int modinv(int a, int m) {
  int b = m, u = 1, v = 0;
  while (b) {
    int t = a / b;
    a -= t * b;
    swap(a, b);
    u -= t * v;
    swap(u, v);
  }
  u %= m;
  if (u < 0)
    u += m;
  return u;
}
int modpow(int a, int n) {
  int res = 1;
  while (n > 0) {
    if (n & 1)
      res = res * a % mod;
    a = a * a % mod;
    n >>= 1;
  }
  return res;
}
// n個を1個以上のx組のグループに分ける重複組み合わせはcom(n-1,x-1)
// グループが0個でもいいときはnにxを足す

int com(int xx, int yy) {
  if (yy == 0) {
    return 1;
  }
  int zz = 1;
  for (int i = xx; i > xx - yy; i--) {
    zz *= i;
    zz %= mod;
  }
  for (int i = 1; i <= yy; i++) {
    zz *= modinv(i, mod);
    zz %= mod;
  }
  return zz;
}
int gcd(int xx, int yy) {
  int p = xx;
  int q = yy;
  if (q > p)
    swap(p, q);
  while (p % q != 0) {
    p %= q;
    swap(p, q);
  }
  return q;
}
int lcm(int xx, int yy) { return xx * yy / gcd(xx, yy); }
bool prime(int xx) {
  int a = xx;
  for (int i = 2; i * i <= xx; i++) {
    if (xx % i == 0) {
      return 0;
    }
  }
  return 1;
}

signed main() {
  int n, m, k, ans = 0;
  cin >> n >> m >> k;
  for (int i = 0; i <= k; i++) {
    ans += ((com(n - 1, i) * m) % mod) * modpow(m - 1, n - i - 1);
    ans %= mod;
  }
  cout << ans << endl;
}