洛谷P7517 数对

传送门

题目描述

给定 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;
}