2017年8月7日提高组T1 呵呵
Description
这天,小A得到了一个序列a[1],a[2]…a[n],他想知道有多少个二元组(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);
}