結果

問題 No.2046 Ans Mod? Mod Ans!
コンテスト
ユーザー QCFium
提出日時 2022-08-19 22:39:02
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 353 ms / 4,000 ms
コード長 1,198 bytes
コンパイル時間 1,919 ms
コンパイル使用メモリ 176,020 KB
実行使用メモリ 6,656 KB
最終ジャッジ日時 2024-12-16 08:24:46
合計ジャッジ時間 3,882 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 18
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>

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

int main() {
	int n = ri();
	int a[n];
	for (auto &i : a) i = ri();
	std::sort(a, a + n);
	
#	define MAX 200000
	int sum[MAX + 2];
	memset(sum, 0, sizeof(sum));
	for (auto i : a) sum[i + 1]++;
	for (int i = 0; i <= MAX; i++) sum[i + 1] += sum[i];
	
	int64_t sum2[MAX + 2];
	memset(sum2, 0, sizeof(sum2));
	for (auto i : a) sum2[i + 1] += i;
	for (int i = 0; i <= MAX; i++) sum2[i + 1] += sum2[i];
	
	std::map<int, int64_t> cache;
	int64_t res = 0;
	for (int i = n; i--; ) {
		int64_t cur = 0;
		if (a[i] <= 500) {
			if (!cache.count(a[i])) {
				for (int j = i + 1; j < n; j++) cur += a[j] % a[i];
				cache[a[i]] = cur;
			} else cur = cache[a[i]];
		} else {
			for (int j = 1; a[i] * j <= MAX; j++) {
				int left = a[i] * j;
				int right = std::min(MAX + 1, a[i] * (j + 1));
				cur += (sum2[right] - sum2[left]) - (int64_t) left * (sum[right] - sum[left]);
			}
		}
		int ok_cnt = sum[MAX + 1] - sum[a[i] + 1];
		res += (int64_t) ok_cnt * a[i] - cur;
		// std::cerr << (int64_t) ok_cnt * a[i] - cur << std::endl;
		// std::cerr << a[i] << " " << cur << std::endl;
	}
	std::cout << res << std::endl;
	
	return 0;
}
0