SSL2667 2017年8月7日提高组T1 呵呵

2017年8月7日提高组T1 呵呵

Description

这天，小A得到了一个序列a[1],a[2]…a
，他想知道有多少个二元组(i,j)满足i!=j且a[i]是a[j]的因数。

Input

第一行有一个正整数n，第二行包含n个整数。

Output

输出一行，表示满足条件的二元组个数。

分析：考虑到跟顺序无关，可以把所有数扔到桶里面，然后对于每一个数，先用组合数统计其本身的贡献，然后暴力查找它的所有倍数即可。

代码

#include <cstdio>
using namespace std;

int n,a[4000005];
long long ans;

int main()
{
scanf("%d",&n);
int max=0;
for (int i=1;i<=n;i++)
{
int x;
scanf("%d",&x);
a[x]++;
if (x>max) max=x;
}
for (int i=1;i<=max;i++)
if (a[i]>0) ans+=a[i]*(a[i]-1);
for (int i=1;i<=max;i++)
if (a[i]>0)
{
int p=0;
for (int j=2*i;j<=max;j+=i)
p+=a[j]*a[i];
ans+=p;
}
printf("%lld",ans);
}