Problem p02793 (C++)

Buggy submission (status: Time Limit Exceeded):
#include <algorithm>
#include <bitset>
#include <cmath>
#include <functional>
#include <iomanip>
#include <iostream>
#include <map>
#include <numeric>
#include <queue>
#include <set>
#include <unordered_set>
#include <vector>

using namespace std;
using ll = long long;
using ull = unsigned long long;
// const ll mod =  998244353 ;
const ll mod = 1000000007;
const ll inf = 1e18;
const int dx[] = {1, -1, 0, 0};
const int dy[] = {0, 0, 1, -1};

#define REP(i, n) for (ll i = 0; i < (n); ++i)
#define REP_FROM(i, j, n) for (ll i = (j); i < (n); ++i)
#define REP_REV(i, n) for (ll i = n - 1; i >= 0; --i)
#define all(x) (x).begin(), (x).end()
#define sz(x) ll(x.size())

template <typename T> void print_vec(vector<T> a) {
  int n = a.size();
  REP(i, n) {
    cout >> a[i];
    if (i != n - 1)
      cout << " ";
    else
      cout << endl;
  }
}
ll power(ll base, ll exponent) {
  if (exponent % 2) {
    return power(base, exponent - 1) * base % mod;
  } else if (exponent) {
    ll root_ans = power(base, exponent / 2);
    return root_ans * root_ans % mod;
  } else {
    return 1;
  }
}
ll inverse(ll x) { return power(x, mod - 2); }
ll gcd(ll a, ll b) {
  if (a < b)
    gcd(b, a);
  ll r;
  while (r = a % b) {
    a = b;
    b = r;
  }
  return b;
}
template <typename T> ll sum(T begin, T end) {
  return accumulate(begin, end, 0ll);
}
struct combination {
  vector<ll> fact, inv;
  combination(int sz) : fact(sz + 1), inv(sz + 1) {
    fact[0] = 1;
    for (int i = 1; i <= sz; i++) {
      fact[i] = fact[i - 1] * i % mod;
    }
    inv[sz] = power(fact[sz], mod - 2);
    for (int i = sz - 1; i >= 0; i--) {
      inv[i] = inv[i + 1] * (i + 1) % mod;
    }
  }
  ll C(int p, int q) const {
    if (q < 0 || p < q)
      return 0;
    return (fact[p] * inv[q] % mod * inv[p - q] % mod);
  }
};

using Pair = pair<ll, ll>;

template <ll Modulus> struct ModInt {
  ll a;
  constexpr ModInt(const ll x = 0) noexcept : a((x % mod + mod) % mod) {}
  constexpr ll &value() noexcept { return a; }
  constexpr const ll &value() const noexcept { return a; }
  constexpr ModInt operator+(const ModInt x) const noexcept {
    return ModInt(*this) += x;
  }
  constexpr ModInt operator-(const ModInt x) const noexcept {
    return ModInt(*this) -= x;
  }
  constexpr ModInt operator*(const ModInt x) const noexcept {
    return ModInt(*this) *= x;
  }
  constexpr ModInt operator/(const ModInt x) const noexcept {
    return ModInt(*this) /= x;
  }
  constexpr ModInt &operator+=(const ModInt x) noexcept {
    a += x.a;
    if (a >= Modulus) {
      a -= Modulus;
    }
    return *this;
  }
  constexpr ModInt &operator-=(const ModInt x) noexcept {
    if (a < x.a) {
      a += Modulus;
    }
    a -= x.a;
    return *this;
  }
  constexpr ModInt &operator*=(const ModInt x) noexcept {
    a = a * x.a % Modulus;
    return *this;
  }
  constexpr ModInt &operator/=(ModInt x) noexcept {
    ll exp = Modulus - 2;
    while (exp) {
      if (exp % 2) {
        *this *= x;
      }
      x *= x;
      exp /= 2;
    }
    return *this;
  }
  constexpr ModInt operator-() noexcept { return ModInt(-a); }
  friend ostream &operator<<(ostream &os, const ModInt &m) {
    os << m.a;
    return os;
  }
};
using modint = ModInt<mod>;

signed main() {
  ios::sync_with_stdio(false);
  cout << fixed << setprecision(15);
  ll n;
  cin >> n;
  vector<ll> a(n);
  vector<map<int, int>> b(n);
  map<int, int> s;
  REP(i, n) cin >> a[i];
  REP(i, n) {
    ll x = a[i];
    for (ll j = 2; j * j <= x; j++) {
      while (x % j == 0) {
        b[i][j]++;
        x /= j;
      }
    }
    if (x != 1)
      b[i][x]++;
  }
  REP(i, n) {
    for (auto &x : b[i]) {
      int y = x.first;
      s[y] = max(s[y], x.second);
    }
  }
  modint ans = 0;
  REP(i, n) {
    modint tmp = 1;
    for (auto &x : s) {
      int y = x.first;
      tmp *= power(y, x.second - b[i][y]);
    }
    ans += tmp;
  }
  cout << ans << endl;
  return 0;
}


Fixed submission:
#include <algorithm>
#include <bitset>
#include <cmath>
#include <functional>
#include <iomanip>
#include <iostream>
#include <map>
#include <numeric>
#include <queue>
#include <set>
#include <unordered_set>
#include <vector>

using namespace std;
using ll = long long;
using ull = unsigned long long;
// const ll mod =  998244353 ;
const ll mod = 1000000007;
const ll inf = 1e18;
const int dx[] = {1, -1, 0, 0};
const int dy[] = {0, 0, 1, -1};

#define REP(i, n) for (ll i = 0; i < (n); ++i)
#define REP_FROM(i, j, n) for (ll i = (j); i < (n); ++i)
#define REP_REV(i, n) for (ll i = n - 1; i >= 0; --i)
#define all(x) (x).begin(), (x).end()
#define sz(x) ll(x.size())

template <typename T> void print_vec(vector<T> a) {
  int n = a.size();
  REP(i, n) {
    cout >> a[i];
    if (i != n - 1)
      cout << " ";
    else
      cout << endl;
  }
}
ll power(ll base, ll exponent) {
  if (exponent % 2) {
    return power(base, exponent - 1) * base % mod;
  } else if (exponent) {
    ll root_ans = power(base, exponent / 2);
    return root_ans * root_ans % mod;
  } else {
    return 1;
  }
}
ll inverse(ll x) { return power(x, mod - 2); }
ll gcd(ll a, ll b) {
  if (a < b)
    gcd(b, a);
  ll r;
  while (r = a % b) {
    a = b;
    b = r;
  }
  return b;
}
template <typename T> ll sum(T begin, T end) {
  return accumulate(begin, end, 0ll);
}
struct combination {
  vector<ll> fact, inv;
  combination(int sz) : fact(sz + 1), inv(sz + 1) {
    fact[0] = 1;
    for (int i = 1; i <= sz; i++) {
      fact[i] = fact[i - 1] * i % mod;
    }
    inv[sz] = power(fact[sz], mod - 2);
    for (int i = sz - 1; i >= 0; i--) {
      inv[i] = inv[i + 1] * (i + 1) % mod;
    }
  }
  ll C(int p, int q) const {
    if (q < 0 || p < q)
      return 0;
    return (fact[p] * inv[q] % mod * inv[p - q] % mod);
  }
};

using Pair = pair<ll, ll>;

template <ll Modulus> struct ModInt {
  ll a;
  constexpr ModInt(const ll x = 0) noexcept : a((x % mod + mod) % mod) {}
  constexpr ll &value() noexcept { return a; }
  constexpr const ll &value() const noexcept { return a; }
  constexpr ModInt operator+(const ModInt x) const noexcept {
    return ModInt(*this) += x;
  }
  constexpr ModInt operator-(const ModInt x) const noexcept {
    return ModInt(*this) -= x;
  }
  constexpr ModInt operator*(const ModInt x) const noexcept {
    return ModInt(*this) *= x;
  }
  constexpr ModInt operator/(const ModInt x) const noexcept {
    return ModInt(*this) /= x;
  }
  constexpr ModInt &operator+=(const ModInt x) noexcept {
    a += x.a;
    if (a >= Modulus) {
      a -= Modulus;
    }
    return *this;
  }
  constexpr ModInt &operator-=(const ModInt x) noexcept {
    if (a < x.a) {
      a += Modulus;
    }
    a -= x.a;
    return *this;
  }
  constexpr ModInt &operator*=(const ModInt x) noexcept {
    a = a * x.a % Modulus;
    return *this;
  }
  constexpr ModInt &operator/=(ModInt x) noexcept {
    ll exp = Modulus - 2;
    while (exp) {
      if (exp % 2) {
        *this *= x;
      }
      x *= x;
      exp /= 2;
    }
    return *this;
  }
  constexpr ModInt operator-() noexcept { return ModInt(-a); }
  friend ostream &operator<<(ostream &os, const ModInt &m) {
    os << m.a;
    return os;
  }
};
using modint = ModInt<mod>;

signed main() {
  ios::sync_with_stdio(false);
  cout << fixed << setprecision(15);
  ll n;
  cin >> n;
  vector<ll> a(n);
  vector<map<int, int>> b(n);
  map<int, int> s;
  REP(i, n) cin >> a[i];
  REP(i, n) {
    ll x = a[i];
    for (ll j = 2; j * j <= x; j++) {
      while (x % j == 0) {
        b[i][j]++;
        x /= j;
      }
    }
    if (x != 1)
      b[i][x]++;
  }
  REP(i, n) {
    for (auto &x : b[i]) {
      int y = x.first;
      s[y] = max(s[y], x.second);
    }
  }
  modint ans = 0;
  REP(i, n) { ans += static_cast<modint>(1) / a[i]; }
  for (auto &x : s) {
    REP(i, x.second) { ans *= x.first; }
  }
  cout << ans << endl;
  return 0;
}