結果
問題 |
No.2046 Ans Mod? Mod Ans!
|
ユーザー |
![]() |
提出日時 | 2022-08-19 22:06:54 |
言語 | C++17 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 50 ms / 4,000 ms |
コード長 | 2,456 bytes |
コンパイル時間 | 1,311 ms |
コンパイル使用メモリ | 117,552 KB |
最終ジャッジ日時 | 2025-01-31 01:11:35 |
ジャッジサーバーID (参考情報) |
judge4 / judge1 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 18 |
ソースコード
#include <iostream> #include <algorithm> #include <iomanip> #include <vector> #include <queue> #include <deque> #include <set> #include <map> #include <tuple> #include <cmath> #include <numeric> #include <functional> #include <cassert> #define debug_value(x) cerr << "line" << __LINE__ << ":<" << __func__ << ">:" << #x << "=" << x << endl; #define debug(x) cerr << "line" << __LINE__ << ":<" << __func__ << ">:" << x << endl; template<class T> inline bool chmax(T& a, T b) { if (a < b) { a = b; return 1; } return 0; } template<class T> inline bool chmin(T& a, T b) { if (a > b) { a = b; return 1; } return 0; } using namespace std; typedef long long ll; template<typename T> vector<vector<T>> vec2d(int n, int m, T v){ return vector<vector<T>>(n, vector<T>(m, v)); } template<typename T> vector<vector<vector<T>>> vec3d(int n, int m, int k, T v){ return vector<vector<vector<T>>>(n, vector<vector<T>>(m, vector<T>(k, v))); } template<typename T> void print_vector(vector<T> v, char delimiter=' '){ if(v.empty()) { cout << endl; return; } for(int i = 0; i+1 < v.size(); i++) cout << v[i] << delimiter; cout << v.back() << endl; } template<typename T> class Cumsum{ public: Cumsum(vector<T> v): v(v){ n = v.size(); v_cumsum = vector<T>(n+1, T(0)); for(int i = 0; i < n; i++) v_cumsum[i+1] = v_cumsum[i]+v[i]; } /** * v[l] + ... + v[r-1] */ T sum(int l, int r){ if(r <= l) return T(0); if(r > n) r = n; if(l < 0) l = 0; return v_cumsum[r]-v_cumsum[l]; } private: int n; vector<T> v; vector<T> v_cumsum; }; const int N = 200000; int main(){ ios::sync_with_stdio(false); cin.tie(0); cout << setprecision(10) << fixed; int n; cin >> n; vector<ll> a(n); vector<ll> cnt(N+1); vector<ll> sum(N+1); for(int i = 0; i < n; i++) { cin >> a[i]; cnt[a[i]]++; sum[a[i]] += a[i]; } auto cumsum = Cumsum<ll>(cnt); auto cumsum_all = Cumsum<ll>(sum); ll ans = 0; for(ll x = 1; x <= N; x++){ ll cnt_all = cumsum.sum(x+1, N+1); ll sum_all = cumsum_all.sum(x+1, N+1); ans += cnt[x]*(cnt_all*x - sum_all); for(ll i = 1; i*x <= N; i++){ int l = i*x, r = (i+1)*x; if(i == 1) l++; ll c = cumsum.sum(l, r); ans += cnt[x]*x*i*c; } } cout << ans << endl; }