#987934 (C++23(gcc13)) No.1233 割り切れない気持ち

提出ソース

結果

問題	No.1233 割り切れない気持ち
ユーザー	ir5
提出日時	2024-06-11 23:37:36
言語	C++23(gcc13) (gcc 13.2.0 + boost 1.83.0)
結果	AC
実行時間	103 ms / 3,153 ms
コード長	2,028 bytes
コンパイル時間	1,543 ms
コンパイル使用メモリ	143,120 KB
実行使用メモリ	8,832 KB
最終ジャッジ日時	2024-06-11 23:37:51
合計ジャッジ時間	5,303 ms
ジャッジサーバーID （参考情報）	judge5 / judge3

このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示

入力	結果	実行時間実行使用メモリ
testcase_00	AC	17 ms 7,936 KB
testcase_01	AC	17 ms 7,936 KB
testcase_02	AC	17 ms 7,936 KB
testcase_03	AC	17 ms 7,936 KB
testcase_04	AC	17 ms 7,936 KB
testcase_05	AC	17 ms 8,064 KB
testcase_06	AC	17 ms 8,064 KB
testcase_07	AC	31 ms 8,064 KB
testcase_08	AC	42 ms 8,192 KB
testcase_09	AC	66 ms 8,576 KB
testcase_10	AC	70 ms 8,576 KB
testcase_11	AC	30 ms 8,192 KB
testcase_12	AC	103 ms 8,576 KB
testcase_13	AC	78 ms 8,704 KB
testcase_14	AC	79 ms 8,704 KB
testcase_15	AC	78 ms 8,704 KB
testcase_16	AC	77 ms 8,704 KB
testcase_17	AC	47 ms 8,704 KB
testcase_18	AC	77 ms 8,576 KB
testcase_19	AC	81 ms 8,704 KB
testcase_20	AC	47 ms 8,576 KB
testcase_21	AC	59 ms 8,704 KB
testcase_22	AC	59 ms 8,704 KB
testcase_23	AC	60 ms 8,704 KB
testcase_24	AC	60 ms 8,576 KB
testcase_25	AC	60 ms 8,704 KB
testcase_26	AC	59 ms 8,704 KB
testcase_27	AC	68 ms 8,832 KB
testcase_28	AC	68 ms 8,704 KB
testcase_29	AC	67 ms 8,576 KB
testcase_30	AC	67 ms 8,832 KB
testcase_31	AC	68 ms 8,704 KB
testcase_32	AC	67 ms 8,704 KB
testcase_33	AC	68 ms 8,704 KB
testcase_34	AC	68 ms 8,704 KB
testcase_35	AC	68 ms 8,832 KB
testcase_36	AC	68 ms 8,704 KB
testcase_37	AC	17 ms 7,936 KB
testcase_38	AC	47 ms 8,704 KB
testcase_39	AC	62 ms 8,576 KB
testcase_40	AC	62 ms 8,704 KB

権限があれば一括ダウンロードができます

ソースコード

raw source code

#include <algorithm>
#include <cassert>
#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <cmath>
#include <iostream>
#include <tuple>
#include <vector>
#include <map>
#include <set>
#include <queue>
#include <ranges>
#include <iomanip>
using namespace std;
using ll = long long;

auto range(int n) { return views::iota(0, n); }

template<class T, class U> ostream& operator<<(ostream& os, const pair<T, U>& p){ return os << "{" << p.first << ", " << p.second << "}"; }
template<typename T> ostream& operator<<(ostream& os, const vector<T>& obj) { os << "{"; for (const auto& e : obj) os << e << ", "; return os << "}"; }
template<typename T> ostream& operator<<(ostream& os, const set<T>& obj) { os << "set{"; for (const auto& e : obj) os << e << ", "; return os << "}"; }
template<typename T, typename U> ostream& operator<<(ostream& os, const map<T, U>& obj) { os << "map{"; for (const auto& e : obj) os << e << ", "; return os << "}"; }

#ifdef ONLINE_JUDGE
#define dump(expr) ;
#else
#define dump(expr) { cerr << "\033[33m#L" << __LINE__ << ": " << expr << "\033[39m" << endl; }
#endif

ll brute(vector<int> vs) {
  ll res = 0;
  int n = vs.size();
  for (int i : range(n))
  for (int j : range(n)) res += vs[i] % vs[j];
  return res;
}

ll solve() {
  int n; cin >> n;
  vector<int> vs(n);
  for (int i : range(n)) cin >> vs[i];

  dump(brute(vs));

  const int M = 200010;
  // const int M = 210;
  vector<ll> ws(M);

  for (int i : range(n)) {
    ws[vs[i]]++;
  }

  vector<ll> ps0(M), ps1(M);
  for (ll i = 1; i < M; ++i) {
    ps0[i] = ps0[i - 1] + ws[i];
    ps1[i] = ps1[i - 1] + i * ws[i];
  }
  dump(ps0)
  dump(ps1)

  ll res = 0;
  for (int j = 1; j < M; ++j) {
    res += ws[j] * ps1[j - 1];

    for (int b = j; b < M; b += j) {
      int e = min(M - 1, b + j - 1);

      ll add = (ps1[e] - ps1[b - 1]) - (ps0[e] - ps0[b - 1]) * (ll)b;
      assert(add >= 0LL);
      res += ws[j] * add;
    }
  }
  return res;
}

int main() {
  cout << fixed << setprecision(12);
  cout << solve() << endl;
}

yukicoder

結果

テストケース

ソースコード