結果

問題 No.1233 割り切れない気持ち
ユーザー k
提出日時 2021-03-22 10:51:59
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 430 ms / 3,153 ms
コード長 814 bytes
コンパイル時間 1,941 ms
コンパイル使用メモリ 198,720 KB
最終ジャッジ日時 2025-01-19 21:01:06
ジャッジサーバーID
（参考情報） 		judge1 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2
other AC * 39
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>

using namespace std;

int main() {
  ios_base::sync_with_stdio(0);
  cin.tie(0);
  
  int n;
  cin >> n;
  vector<long long> a(n);
  for (int i = 0; i < n; i++)
    cin >> a[i];

  sort(a.begin(), a.end());
  
  vector<long long> pref(n + 1);
  for (int i = 0; i < n; i++)
    pref[i+1] = pref[i] + a[i];
  
  long long ret = 0;
  for (int i = 0; i < n; ) {
    int j = i;
    while (j < n && a[j] == a[i])
      ++j;
    long long x = pref[i];
    for (long long s = a[i]; s <= a[n-1]; s += a[i]) {
      long long t = s + a[i];
      int p = lower_bound(a.begin(), a.end(), s) - a.begin();
      int q = lower_bound(a.begin(), a.end(), t) - a.begin();
      x += (pref[q] - pref[p]) - (q - p) * s;
    }
    ret += (j - i) * x;
    i = j;
  }
  
  cout << ret << endl;
  
  return 0;
}
0