洛谷 P7517 [省选联考 2021 B 卷] 数对

edward1346

于 2024-10-06 09:42:09 发布

阅读量261

点赞数 8

文章标签：算法 c++

本文链接：https://blog.csdn.net/2403_87021226/article/details/142723224

版权

题目传送门

解题思路

其实你只要知道：

$n \times \sum _{i=1} ^{n/i} = n \log n$

这题你就秒了。

我们发现 $a_i \leq 5 \times 10^5$ ，于是开一个桶来统计每个数出现的数量。

我们只需要枚举每一个数的倍数，然后统计。

最后，如果一个数出现了多次，再特判一下即可。

代码

#include<bits/stdc++.h>
using namespace std;

int n;
int cnt[500001];
int mx;
long long ans;
long long temp;
int main()
{
	ios::sync_with_stdio(0);
	cin.tie(0);
	cout.tie(0);
	cin>>n;
	int x;
	for(int i=1;i<=n;i++)
	{
		cin>>x;
		mx=max(mx,x);
		cnt[x]++;
	}
	
	for(int i=1;i<=mx;i++)
	{
		temp=0;
		for(int j=2;j<=mx/i;j++)
		{
			temp+=cnt[i*j];
		}
		temp+=cnt[i]-1;
		temp*=cnt[i];
		ans+=temp;
	}
	cout<<ans;
	return 0;
}