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

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
#define rep(i, n) for (long long i = 0; i < n; i++)
#define repr(i, n, m) for (long long i = m; i < n; i++)
#define mod 1000000007
#define sorti(x) sort(x.begin(), x.end())
#define sortd(x) sort(x.begin(), x.end(), std::greater<long long>())
#define debug(x) std::cerr << (x) << std::endl;
#define roll(x)                                                                \\
  for (auto &&itr : x) {                                                       \\
    cerr << (itr) << " ";                                                      \\
  }

template <class T> inline void chmax(T &ans, T t) {
  if (t > ans)
    ans = t;
}
template <class T> inline void chmin(T &ans, T t) {
  if (t < ans)
    ans = t;
}

template <class T> T inf = numeric_limits<T>::max();

// 方針
// 累積和を取る
// ex) 18171
// s[4] 1
// s[3] 71
// s[2] 171
// s[1] 8171
// s[0] 18171
// mod 2019 = 0の区間(i, j), i < j は, (s[i] - s[j]) mod 2019 = 0
// つまり, s[i] mod 2019 == s[j] mod 2019

ll calc(ll t) { return t * (t - 1) / 2; }

int main() {
  cin.tie(0);
  ios::sync_with_stdio(false);

  string s;
  cin >> s;

  reverse(s.begin(), s.end());
  vector<ll> val(2019, 0);
  val[0] += 1;
  ll fac = 1;
  ll cur = 0;
  for (auto itr : s) {
    ll num = itr - '0';
    cur = (cur + num * fac) % 2019;
    val[cur] += 1;
    fac = (fac * 10) % 2019;
  }

  ll ans = 0;
  for (int i = 0; i < 2019; ++i) {
    ans += calc(val[i]);
  }

  cout << ans << endl;
}


Error: TLE

Fixed submission:

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

#include <boost/multiprecision/cpp_dec_float.hpp>
#include <boost/multiprecision/cpp_int.hpp>
namespace mp = boost::multiprecision;
// 任意長整数型
using Bint = mp::cpp_int;
// 仮数部が1024ビットの浮動小数点数型(TLEしたら小さくする)
using Real = mp::number<mp::cpp_dec_float<1024>>;

#define inf 10e17
#define rep(i, n) for (long long i = 0; i < n; i++)
#define repr(i, n, m) for (long long i = m; i < n; i++)
#define mod 1000000007
#define sorti(x) sort(x.begin(), x.end())
#define sortd(x) sort(x.begin(), x.end(), std::greater<long long>())
#define debug(x) std::cerr << (x) << std::endl;
#define roll(x)                                                                \\
  for (auto &&itr : x) {                                                       \\
    cerr << (itr) << " ";                                                      \\
  }

template <class T> inline void chmax(T &ans, T t) {
  if (t > ans)
    ans = t;
}
template <class T> inline void chmin(T &ans, T t) {
  if (t < ans)
    ans = t;
}
// BEGIN CUT HERE
struct bigint {
  using ll = long long;
  using vll = vector<ll>;

  constexpr static ll base = 1000000000;
  constexpr static ll base_digits = 9;

  vll a;
  ll sign;

  bigint() : sign(1) {}

  bigint(ll v) { *this = v; }

  bigint(const string &s) { read(s); }

  static bigint add_identity() { return bigint(0); }
  static bigint mul_identity() { return bigint(1); }

  void operator=(ll v) {
    sign = 1;
    if (v < 0)
      sign = -1, v = -v;
    for (; v > 0; v = v / base)
      a.push_back(v % base);
  }

  bigint operator+(const bigint &v) const {
    if (sign == v.sign) {
      bigint res = v;

      for (ll i = 0, carry = 0; i < (ll)max(a.size(), v.a.size()) || carry;
           ++i) {
        if (i == (ll)res.a.size())
          res.a.push_back(0);
        res.a[i] += carry + (i < (ll)a.size() ? a[i] : 0);
        carry = res.a[i] >= base;
        if (carry)
          res.a[i] -= base;
      }
      return res;
    }
    return *this - (-v);
  }

  bigint operator-(const bigint &v) const {
    if (sign == v.sign) {
      if (abs() >= v.abs()) {
        bigint res = *this;
        for (ll i = 0, carry = 0; i < (ll)v.a.size() || carry; ++i) {
          res.a[i] -= carry + (i < (ll)v.a.size() ? v.a[i] : 0);
          carry = res.a[i] < 0;
          if (carry)
            res.a[i] += base;
        }
        res.trim();
        return res;
      }
      return -(v - *this);
    }
    return *this + (-v);
  }

  void operator*=(ll v) {
    if (v < 0)
      sign = -sign, v = -v;
    for (ll i = 0, carry = 0; i < (ll)a.size() || carry; ++i) {
      if (i == (ll)a.size())
        a.push_back(0);
      ll cur = a[i] * (ll)v + carry;
      carry = (ll)(cur / base);
      a[i] = (ll)(cur % base);
      // asm("divl %%ecx" : "=a"(carry),"=d"(a[i]) : "A"(cur),"c"(base));
    }
    trim();
  }

  bigint operator*(ll v) const {
    bigint res = *this;
    res *= v;
    return res;
  }

  friend pair<bigint, bigint> divmod(const bigint &a1, const bigint &b1) {
    ll norm = base / (b1.a.back() + 1);
    bigint a = a1.abs() * norm;
    bigint b = b1.abs() * norm;
    bigint q, r;
    q.a.resize(a.a.size());

    for (ll i = a.a.size() - 1; i >= 0; i--) {
      r *= base;
      r += a.a[i];
      ll s1 = r.a.size() <= b.a.size() ? 0 : r.a[b.a.size()];
      ll s2 = r.a.size() <= b.a.size() - 1 ? 0 : r.a[b.a.size() - 1];
      ll d = ((ll)base * s1 + s2) / b.a.back();
      r -= b * d;
      while (r < 0)
        r += b, --d;
      q.a[i] = d;
    }

    q.sign = a1.sign * b1.sign;
    r.sign = a1.sign;
    q.trim();
    r.trim();
    return make_pair(q, r / norm);
  }

  bigint operator/(const bigint &v) const { return divmod(*this, v).first; }

  bigint operator%(const bigint &v) const { return divmod(*this, v).second; }

  void operator/=(ll v) {
    if (v < 0)
      sign = -sign, v = -v;
    for (ll i = (ll)a.size() - 1, rem = 0; i >= 0; --i) {
      ll cur = a[i] + rem * (ll)base;
      a[i] = (ll)(cur / v);
      rem = (ll)(cur % v);
    }
    trim();
  }

  bigint operator/(ll v) const {
    bigint res = *this;
    res /= v;
    return res;
  }

  ll operator%(ll v) const {
    if (v < 0)
      v = -v;
    ll m = 0;
    for (ll i = a.size() - 1; i >= 0; --i)
      m = (a[i] + m * (ll)base) % v;
    return m * sign;
  }

  void operator+=(const bigint &v) { *this = *this + v; }

  void operator-=(const bigint &v) { *this = *this - v; }

  void operator*=(const bigint &v) { *this = *this * v; }

  void operator/=(const bigint &v) { *this = *this / v; }

  bool operator<(const bigint &v) const {
    if (sign != v.sign)
      return sign < v.sign;
    if (a.size() != v.a.size())
      return a.size() * sign < v.a.size() * v.sign;
    for (ll i = a.size() - 1; i >= 0; i--)
      if (a[i] != v.a[i])
        return a[i] * sign < v.a[i] * sign;
    return false;
  }

  bool operator>(const bigint &v) const { return v < *this; }

  bool operator<=(const bigint &v) const { return !(v < *this); }

  bool operator>=(const bigint &v) const { return !(*this < v); }

  bool operator==(const bigint &v) const {
    return !(*this < v) && !(v < *this);
  }

  bool operator!=(const bigint &v) const { return *this < v || v < *this; }

  void trim() {
    while (!a.empty() && !a.back())
      a.pop_back();
    if (a.empty())
      sign = 1;
  }

  bool isZero() const { return a.empty() || (a.size() == 1 && !a[0]); }

  bigint operator-() const {
    bigint res = *this;
    res.sign = -sign;
    return res;
  }

  bigint abs() const {
    bigint res = *this;
    res.sign *= res.sign;
    return res;
  }

  ll longValue() const {
    ll res = 0;
    for (ll i = a.size() - 1; i >= 0; i--)
      res = res * base + a[i];
    return res * sign;
  }

  friend bigint gcd(const bigint &a, const bigint &b) {
    return b.isZero() ? a : gcd(b, a % b);
  }

  friend bigint lcm(const bigint &a, const bigint &b) {
    return a / gcd(a, b) * b;
  }

  void read(const string &s) {
    sign = 1;
    a.clear();
    ll pos = 0;
    while (pos < (ll)s.size() && (s[pos] == '-' || s[pos] == '+')) {
      if (s[pos] == '-')
        sign = -sign;
      ++pos;
    }
    for (ll i = s.size() - 1; i >= pos; i -= base_digits) {
      ll x = 0;
      for (ll j = max(pos, i - base_digits + 1); j <= i; j++)
        x = x * 10 + s[j] - '0';
      a.push_back(x);
    }
    trim();
  }

  friend istream &operator>>(istream &stream, bigint &v) {
    string s;
    stream >> s;
    v.read(s);
    return stream;
  }

  friend ostream &operator<<(ostream &stream, const bigint &v) {
    if (v.sign == -1)
      stream << '-';
    stream << (v.a.empty() ? 0 : v.a.back());
    for (ll i = (ll)v.a.size() - 2; i >= 0; --i)
      stream << setw(base_digits) << setfill('0') << v.a[i];
    return stream;
  }

  static vll convert_base(const vll &a, ll old_digits, ll new_digits) {
    vll p(max(old_digits, new_digits) + 1);
    p[0] = 1;
    for (ll i = 1; i < (ll)p.size(); i++)
      p[i] = p[i - 1] * 10;
    vll res;
    ll cur = 0;
    ll cur_digits = 0;
    for (ll i = 0; i < (ll)a.size(); i++) {
      cur += a[i] * p[cur_digits];
      cur_digits += old_digits;
      while (cur_digits >= new_digits) {
        res.push_back(signed(cur % p[new_digits]));
        cur /= p[new_digits];
        cur_digits -= new_digits;
      }
    }
    res.push_back((signed)cur);
    while (!res.empty() && !res.back())
      res.pop_back();
    return res;
  }

  static vll karatsubaMultiply(const vll &a, const vll &b) {
    ll n = a.size();
    vll res(n + n);
    if (n <= 32) {
      for (ll i = 0; i < n; i++)
        for (ll j = 0; j < n; j++)
          res[i + j] += a[i] * b[j];
      return res;
    }

    ll k = n >> 1;
    vll a1(a.begin(), a.begin() + k);
    vll a2(a.begin() + k, a.end());
    vll b1(b.begin(), b.begin() + k);
    vll b2(b.begin() + k, b.end());

    vll a1b1 = karatsubaMultiply(a1, b1);
    vll a2b2 = karatsubaMultiply(a2, b2);

    for (ll i = 0; i < k; i++)
      a2[i] += a1[i];
    for (ll i = 0; i < k; i++)
      b2[i] += b1[i];

    vll r = karatsubaMultiply(a2, b2);
    for (ll i = 0; i < (ll)a1b1.size(); i++)
      r[i] -= a1b1[i];
    for (ll i = 0; i < (ll)a2b2.size(); i++)
      r[i] -= a2b2[i];

    for (ll i = 0; i < (ll)r.size(); i++)
      res[i + k] += r[i];
    for (ll i = 0; i < (ll)a1b1.size(); i++)
      res[i] += a1b1[i];
    for (ll i = 0; i < (ll)a2b2.size(); i++)
      res[i + n] += a2b2[i];
    return res;
  }

  bigint operator*(const bigint &v) const {
    constexpr static ll nbase = 10000;
    constexpr static ll nbase_digits = 4;

    vll a = convert_base(this->a, base_digits, nbase_digits);
    vll b = convert_base(v.a, base_digits, nbase_digits);

    //*/
    while (a.size() < b.size())
      a.push_back(0);
    while (b.size() < a.size())
      b.push_back(0);
    while (a.size() & (a.size() - 1))
      a.push_back(0), b.push_back(0);
    vll c = karatsubaMultiply(a, b);
    /*/
    if(a.empty()) a.push_back(0);
    if(b.empty()) b.push_back(0);
    vll c=FFT::multiply(a,b);
    //*/

    bigint res;
    res.sign = sign * v.sign;
    for (ll i = 0, carry = 0; i < (ll)c.size(); i++) {
      ll cur = c[i] + carry;
      res.a.push_back((ll)(cur % nbase));
      carry = (ll)(cur / nbase);
      if (i + 1 == (int)c.size() && carry > 0)
        c.push_back(0);
    }

    res.a = convert_base(res.a, nbase_digits, base_digits);
    res.trim();
    return res;
  }
};
int main() {
  string s;
  cin >> s;

  ll ans = 0;
  for (int i = s.size() - 1; i > 0; --i) {
    bigint a = 0;
    bigint t = 1, ten = 10, tw = 2019;
    for (int j = i; j >= 0; --j) {
      bigint tt = s[j] - '0';
      a += tt * t;
      t *= ten;
      //    cerr << a << endl;
      if (a % tw == 0) {
        // cout << j << " " << i << " " << a << endl;
        ans += 1;
        t = 1;
        a = 0;
      }
    }
  }
  cout << ans << endl;
}