题目描述
给定 nn 个正整数 a_ia
i
,请你求出有多少个数对 (i, j)(i,j) 满足 1 \le i \le n1≤i≤n,1 \le j \le n1≤j≤n,i \ne ji
=j 且 a_ia
i
是 a_ja
j
的倍数。
输入格式
第一行,一个整数 nn,表示数字个数。
第二行,nn 个整数,表示 a_ia
i
。
输出格式
输出一行,一个整数,表示答案。
输入输出样例
输入 #1复制
6
16 11 6 1 9 11
输出 #1复制
7
说明/提示
对于 40 %40% 的数据,n \le 1000n≤1000。
对于 70 %70% 的数据,1 \le a_i \le 5 \times {10}^31≤a
i
≤5×10
3
。
对于 100 %100% 的数据,2 \le n \le 2 \times {10}^52≤n≤2×10
5
,1 \le a_i \le 5 \times {10}^51≤a
i
≤5×10
5
。
上代码:
#include <iostream>
#include <cstdio>
#include <algorithm>
using namespace std;
int n,a,maxn;
long long ans,t[500005];
int main ()
{
scanf ("%d",&n);
for(int i=1;i<=n;++i)
{
scanf ("%d",&a);
++t[a];
maxn=max(maxn,a);
}
for(int i=1;i<=maxn;++i)
{
if (t[(int)i])
{
if (i==1) ans+=(t[i]*(n-1));
else
{
long long x=(long long)i*2;
ans+=(t[i]*(t[i]-1));
while (x<=maxn)
{
ans+=(t[i]*t[x]);
x+=(long long)i;
}
}
}
}
printf("%lld\n",ans);
return 0;
}