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

#include "bits/stdc++.h"

using namespace std;
typedef vector<int> VI;
typedef vector<VI> VVI;
typedef vector<VVI> VVVI;
typedef vector<string> VS;
typedef pair<int, int> PII;
typedef vector<PII> VPII;
typedef long long LL;
typedef priority_queue<int> PQ_DESC;
typedef priority_queue<int, vector<int>, greater<int>> PQ_ASC;
typedef priority_queue<PII> PQ_DESC_PII;
typedef priority_queue<PII, vector<PII>, greater<PII>> PQ_ASC_PII;
typedef vector<LL> VLL;
typedef vector<VLL> VVLL;
typedef vector<VVLL> VVVLL;

#define SORT_ASC(c) sort((c).begin(), (c).end())
#define SORT_DESC(c)                                                           \\
  sort((c).begin(), (c).end(), greater<typeof((c).begin())>())

#define FOR(i, a, b) for (int i = (a); i < (b); ++i)
#define REP(i, n) FOR(i, 0, n)
#define FORL(i, a, b) for (LL i = (a); i < (b); ++i)
#define REPL(i, n) FORL(i, 0, n)
#define SIZE(a) int((a).size())
#define ALL(a) (a).begin(), (a).end()

const double EPS = 1e-10;
const double PI = acos(-1.0);

// debug func
template <typename T> void vprint(vector<T> v) {
  for (auto x : v) {
    cerr << x << " ";
  }
  cerr << endl;
}

template <typename T> void vvprint(vector<vector<T>> vv) {
  REP(i, SIZE(vv)) {
    REP(j, SIZE(vv[i])) { cerr << vv[i][j] << " "; }
    cerr << endl;
  }
}

template <typename Iterator>
inline bool next_combination(const Iterator first, Iterator k,
                             const Iterator last) {
  /* Credits: Thomas Draper */
  if ((first == last) || (first == k) || (last == k))
    return false;
  Iterator itr1 = first;
  Iterator itr2 = last;
  ++itr1;
  if (last == itr1)
    return false;
  itr1 = last;
  --itr1;
  itr1 = k;
  --itr2;
  while (first != itr1) {
    if (*--itr1 < *itr2) {
      Iterator j = k;
      while (!(*itr1 < *j))
        ++j;
      iter_swap(itr1, j);
      ++itr1;
      ++j;
      itr2 = k;
      rotate(itr1, j, last);
      while (last != j) {
        ++j;
        ++itr2;
      }
      rotate(k, itr2, last);
      return true;
    }
  }
  rotate(first, k, last);
  return false;
}

inline double get_time_sec(void) {
  return static_cast<double>(chrono::duration_cast<chrono::nanoseconds>(
                                 chrono::steady_clock::now().time_since_epoch())
                                 .count()) /
         1000000000;
}

template <typename T> T gcd(T a, T b) {
  if (a > b)
    swap(a, b);
  if (a == 0)
    return b;
  else
    return gcd(b % a, a);
}

template <typename T> T lcm(T a, T b) { return (a / gcd(a, b)) * b; }

template <typename T> map<T, T> prime_list(T n) {
  map<T, T> ret;
  for (T i = 2; i * i <= n; i++) {
    if (n % i == 0) {
      ret[i] = 0;
      while (n % i == 0) {
        n /= i;
        ret[i]++;
      }
    }
  }
  if (n != 1)
    ret[n]++;
  return ret;
}

#define MOD 2019

LL mypow(LL a, LL n) {
  if (n == 1)
    return a % MOD;
  if (n % 2 == 1)
    return (a * mypow(a, n - 1)) % MOD;
  LL t = mypow(a, n / 2);
  return (t * t) % MOD;
}

int main(void) {
  string s;
  cin >> s;
  int n = s.size();
  VI dp2(2019, 0);
  int m = (int)(s[n - 1] - '0');
  dp2[m]++;
  REP(i, n - 1) {
    m = (m + mypow(10, i + 1) * (int)(s[n - i - 2] - '0')) % MOD;
    dp2[m]++;
  }
  // vprint(dp);
  // vprint(dp2);
  LL ans = dp2[0] * (dp2[0] + 1) / 2;
  FOR(i, 1, 2019) ans += dp2[i] * (dp2[i] - 1) / 2;
  cout << ans << endl;
}

Error: TLE

Fixed submission:

#include "bits/stdc++.h"

using namespace std;
typedef vector<int> VI;
typedef vector<VI> VVI;
typedef vector<VVI> VVVI;
typedef vector<string> VS;
typedef pair<int, int> PII;
typedef vector<PII> VPII;
typedef long long LL;
typedef priority_queue<int> PQ_DESC;
typedef priority_queue<int, vector<int>, greater<int>> PQ_ASC;
typedef priority_queue<PII> PQ_DESC_PII;
typedef priority_queue<PII, vector<PII>, greater<PII>> PQ_ASC_PII;
typedef vector<LL> VLL;
typedef vector<VLL> VVLL;
typedef vector<VVLL> VVVLL;

#define SORT_ASC(c) sort((c).begin(), (c).end())
#define SORT_DESC(c)                                                           \\
  sort((c).begin(), (c).end(), greater<typeof((c).begin())>())

#define FOR(i, a, b) for (int i = (a); i < (b); ++i)
#define REP(i, n) FOR(i, 0, n)
#define FORL(i, a, b) for (LL i = (a); i < (b); ++i)
#define REPL(i, n) FORL(i, 0, n)
#define SIZE(a) int((a).size())
#define ALL(a) (a).begin(), (a).end()

const double EPS = 1e-10;
const double PI = acos(-1.0);

// debug func
template <typename T> void vprint(vector<T> v) {
  for (auto x : v) {
    cerr << x << " ";
  }
  cerr << endl;
}

template <typename T> void vvprint(vector<vector<T>> vv) {
  REP(i, SIZE(vv)) {
    REP(j, SIZE(vv[i])) { cerr << vv[i][j] << " "; }
    cerr << endl;
  }
}

template <typename Iterator>
inline bool next_combination(const Iterator first, Iterator k,
                             const Iterator last) {
  /* Credits: Thomas Draper */
  if ((first == last) || (first == k) || (last == k))
    return false;
  Iterator itr1 = first;
  Iterator itr2 = last;
  ++itr1;
  if (last == itr1)
    return false;
  itr1 = last;
  --itr1;
  itr1 = k;
  --itr2;
  while (first != itr1) {
    if (*--itr1 < *itr2) {
      Iterator j = k;
      while (!(*itr1 < *j))
        ++j;
      iter_swap(itr1, j);
      ++itr1;
      ++j;
      itr2 = k;
      rotate(itr1, j, last);
      while (last != j) {
        ++j;
        ++itr2;
      }
      rotate(k, itr2, last);
      return true;
    }
  }
  rotate(first, k, last);
  return false;
}

inline double get_time_sec(void) {
  return static_cast<double>(chrono::duration_cast<chrono::nanoseconds>(
                                 chrono::steady_clock::now().time_since_epoch())
                                 .count()) /
         1000000000;
}

template <typename T> T gcd(T a, T b) {
  if (a > b)
    swap(a, b);
  if (a == 0)
    return b;
  else
    return gcd(b % a, a);
}

template <typename T> T lcm(T a, T b) { return (a / gcd(a, b)) * b; }

template <typename T> map<T, T> prime_list(T n) {
  map<T, T> ret;
  for (T i = 2; i * i <= n; i++) {
    if (n % i == 0) {
      ret[i] = 0;
      while (n % i == 0) {
        n /= i;
        ret[i]++;
      }
    }
  }
  if (n != 1)
    ret[n]++;
  return ret;
}

#define MOD 1000000007

LL mypow(LL a, LL n) {
  if (n == 1)
    return a % MOD;
  if (n % 2 == 1)
    return (a * mypow(a, n - 1)) % MOD;
  LL t = mypow(a, n / 2);
  return (t * t) % MOD;
}

int main(void) {
  string s;
  cin >> s;
  int n = s.size();
  LL ans = 0;
  REP(i, n - 3) {
    // cerr << s.substr(i, 4) << endl;
    int m = stoi(s.substr(i, 4)) % 2019;
    if (m == 0)
      ans++;
    FOR(j, i + 4, n) {
      m = (m * 10 + (int)(s[j] - '0')) % 2019;
      if (m == 0)
        ans++;
    }
  }
  cout << ans << endl;
}