結果

問題 No.1233 割り切れない気持ち
ユーザー HaarHaar
提出日時 2020-09-18 22:29:18
言語 C++17
(gcc 13.2.0 + boost 1.83.0)
結果
AC  
実行時間 25 ms / 3,153 ms
コード長 3,428 bytes
コンパイル時間 2,175 ms
コンパイル使用メモリ 203,232 KB
実行使用メモリ 6,452 KB
最終ジャッジ日時 2023-09-04 21:50:54
合計ジャッジ時間 4,705 ms
ジャッジサーバーID
（参考情報） 		judge12 / judge11
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 4 ms
4,652 KB
testcase_01 AC 3 ms
4,572 KB
testcase_02 AC 4 ms
4,644 KB
testcase_03 AC 5 ms
4,576 KB
testcase_04 AC 4 ms
4,576 KB
testcase_05 AC 4 ms
4,632 KB
testcase_06 AC 4 ms
4,684 KB
testcase_07 AC 10 ms
5,172 KB
testcase_08 AC 13 ms
5,432 KB
testcase_09 AC 21 ms
5,924 KB
testcase_10 AC 22 ms
5,896 KB
testcase_11 AC 9 ms
4,852 KB
testcase_12 AC 25 ms
6,288 KB
testcase_13 AC 24 ms
6,276 KB
testcase_14 AC 25 ms
6,264 KB
testcase_15 AC 25 ms
6,284 KB
testcase_16 AC 25 ms
6,276 KB
testcase_17 AC 16 ms
6,224 KB
testcase_18 AC 24 ms
6,272 KB
testcase_19 AC 22 ms
6,376 KB
testcase_20 AC 15 ms
6,156 KB
testcase_21 AC 17 ms
6,204 KB
testcase_22 AC 18 ms
6,176 KB
testcase_23 AC 18 ms
6,236 KB
testcase_24 AC 18 ms
6,236 KB
testcase_25 AC 17 ms
6,220 KB
testcase_26 AC 18 ms
6,268 KB
testcase_27 AC 19 ms
6,452 KB
testcase_28 AC 19 ms
6,280 KB
testcase_29 AC 19 ms
6,188 KB
testcase_30 AC 19 ms
6,280 KB
testcase_31 AC 19 ms
6,212 KB
testcase_32 AC 19 ms
6,284 KB
testcase_33 AC 19 ms
6,156 KB
testcase_34 AC 19 ms
6,280 KB
testcase_35 AC 20 ms
6,172 KB
testcase_36 AC 20 ms
6,224 KB
testcase_37 AC 3 ms
4,656 KB
testcase_38 AC 15 ms
6,156 KB
testcase_39 AC 20 ms
6,204 KB
testcase_40 AC 21 ms
6,184 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>

#ifdef DEBUG
#include <Mylib/Debug/debug.cpp>
#else
#define dump(...) ((void)0)
#endif

template <typename T, typename U>
bool chmin(T &a, const U &b){
  return (a > b ? a = b, true : false);
}

template <typename T, typename U>
bool chmax(T &a, const U &b){
  return (a < b ? a = b, true : false);
}

template <typename T, size_t N, typename U>
void fill_array(T (&a)[N], const U &v){
  std::fill((U*)a, (U*)(a + N), v);
}

template <typename T, size_t N, size_t I = N>
auto make_vector(const std::array<int, N> &a, T value = T()){
  static_assert(I >= 1);
  static_assert(N >= 1);
  if constexpr (I == 1){
    return std::vector<T>(a[N - I], value);
  }else{
    return std::vector(a[N - I], make_vector<T, N, I - 1>(a, value));
  }
}

template <typename T>
std::ostream& operator<<(std::ostream &s, const std::vector<T> &a){
  for(auto it = a.begin(); it != a.end(); ++it){
    if(it != a.begin()) s << " ";
    s << *it;
  }
  return s;
}

template <typename T>
std::istream& operator>>(std::istream &s, std::vector<T> &a){
  for(auto &x : a) s >> x;
  return s;
}

namespace haar_lib {
  template <typename T, typename Add = std::plus<T>, typename Minus = std::minus<T>>
  class cumulative_sum_1d {
    std::vector<T> data;
    const int N;
    const Add add;
    const Minus minus;
    bool is_built = false;

  public:
    cumulative_sum_1d(const std::vector<T> &a, const T &e = 0, const Add &add = Add(), const Minus &minus = Minus()):
      N(a.size()), add(add), minus(minus)
    {
      data.assign(N + 1, e);
      for(int i = 0; i < N; ++i) data[i + 1] = a[i];
    }

    cumulative_sum_1d(int N, const T &e = 0, const Add &add = Add(), const Minus &minus = Minus()):
      N(N), add(add), minus(minus)
    {
      data.assign(N + 1, e);
    }

    auto& update(int i, const T &val){
      assert(not is_built);
      data[i + 1] = add(data[i + 1], val);
      return *this;
    }

    auto& build(){
      assert(not is_built);
      for(int i = 0; i < N; ++i) data[i + 1] = add(data[i + 1], data[i]);
      is_built = true;
      return *this;
    }

    /**
     * @attention [i, j)
     */
    T get(int i, int j) const {
      assert(is_built);
      return minus(data[j], data[i]);
    }
  };
}



namespace haar_lib {}

namespace solver {
  using namespace haar_lib;

  constexpr int m1000000007 = 1000000007;
  constexpr int m998244353 = 998244353;

  void init(){
    std::cin.tie(0);
    std::ios::sync_with_stdio(false);
    std::cout << std::fixed << std::setprecision(12);
    std::cerr << std::fixed << std::setprecision(12);
    std::cin.exceptions(std::ios_base::failbit);
  }

  constexpr int MAX = 200000;

  void solve(){
    int N; std::cin >> N;
    std::vector<int64_t> A(N); std::cin >> A;

    std::vector<int> count(MAX + 1);
    for(auto x : A) count[x] += 1;
    cumulative_sum_1d cs(count);
    cs.build();

    int64_t sum = std::accumulate(A.begin(), A.end(), 0LL);

    int64_t ans = 0;
    dump(cs.get(1, MAX + 1));
    for(int i = 1; i <= MAX; ++i){
      if(count[i] == 0) continue;
      ans += sum * count[i];
      dump(ans);
      for(int j = i; j <= MAX; j += i){
        ans -= (int64_t)i * cs.get(j, MAX + 1) * count[i];
      }
      dump(ans);
    }

    std::cout << ans << "\n";
  }
}

int main(){
  solver::init();
  while(true){
    try{
      solver::solve();
    }catch(const std::istream::failure &e){
      break;
    }
  }
  return 0;
}
0