結果

問題 No.1233 割り切れない気持ち
ユーザー QCFium
提出日時 2019-07-31 00:52:43
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 2,425 ms / 3,153 ms
コード長 1,511 bytes
コンパイル時間 1,721 ms
コンパイル使用メモリ 182,444 KB
実行使用メモリ 17,648 KB
最終ジャッジ日時 2024-06-22 19:09:20
合計ジャッジ時間 23,666 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2
other AC * 39
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>

int ri() {
	int n;
	scanf("%d", &n);
	return n;
}

int main() {
	int n = ri();
	std::map<int, int> a;
	std::vector<std::tuple<int, int, int64_t> > sum;
	for (int i = 0; i < n; i++) a[ri()]++;
	std::pair<int, int64_t> cur = {0, 0};
	for (auto &i : a) {
		sum.push_back(std::tuple<int, int, int64_t> (i.first, cur.first, cur.second));
		cur.first += i.second;
		cur.second += (int64_t) i.first * i.second;
	}
	
	int cache[801];
	for (int i = 0; i <= 800; i++) cache[i] = -1;
	int64_t ans = 0;
	for (auto &i : a) {
		int64_t res = 0;
		if (i.first <= 800) {
			if (cache[i.first] == -1) {
				cache[i.first] = 0;
				for (auto &j : a) cache[i.first] += (j.first % i.first) * j.second;
			}
			res = cache[i.first];
		} else {
			for (int j = 0; j < 200000; j += i.first) {
				auto low = std::lower_bound(sum.begin(), sum.end(), std::tuple<int, int, int64_t>(j, 0, 0));
				auto high = std::lower_bound(sum.begin(), sum.end(), std::tuple<int, int, int64_t>(j + i.first, 0, 0));
				int64_t xsum = (high == sum.end() ? cur.second : std::get<2>(*high)) -
							   (low == sum.end() ? cur.second : std::get<2>(*low));
				int num = (high == sum.end() ? cur.first : std::get<1>(*high)) -
						  (low == sum.end() ? cur.first : std::get<1>(*low));
				// std::cerr << xsum << " " << num * i.first * (j / i.first) << std::endl;
				res += xsum;
				res -= (int64_t) num * i.first * (j / i.first);
			}
		}
		res *= i.second;
		ans += res;
	}
	std::cout << ans << std::endl;
	
    return 0;
}
0