zzulioj 1902: 985的因子对难题 [思维]

1902: 985的因子对难题

Time Limit: 2 Sec   Memory Limit: 128 MB
Submit: 142   Solved: 38

Submit Status Web Board

Description

985有n个正整数，他想知道存在多少个不同的因子对(a[i], a[j])使得

1 <= i, j <= n && i != j && a[j] % a[i] == 0，其中i和j是元素的下标。

特别地，他认为（a[i]，a[j]）与（a[j]，a[i]）是一样的因子对。

Input

第一行输入一个整数t，代表有t组测试数据。

每组数据占两行，第一行输入一个n代表元素个数，下面一行输入n个整数a[]。

注：1 <= t <= 30，1 <= n <= 1e5，1 <= a[] <= 1e6。

Output

一个整数代表最后的答案。

Sample Input

2
5
1 2 3 4 5
5
2 2 2 2 2

Sample Output

5
10

去重：记num[a]为a出现的次数；
求因子数：记sum[a]为a的因子数；（打表） 
num[a]>1时，s+=num[i]*(num[i]-1)/2;
s+=sum[i]*num[i]

思维。。。。。。

代码：

#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
int num[1000010],sum[1000010];
int main()
{
	int t,n,a;	
	int i,j;
	int maxn;
	scanf("%d",&t);
	while(t--)
	{
		memset(num,0,sizeof(num));
		memset(sum,0,sizeof(sum));
		maxn=0;
		scanf("%d",&n);
		for(i=0;i<n;i++)
		{
			scanf("%d",&a);
			num[a]++;
			maxn=max(maxn,a);
		}	
		for(i=1;i<=maxn;i++)
		{
			for(j=2*i;j<=maxn;j+=i)
			{
				if(num[i])
					sum[j]+=num[i];				
			}
		}
		long long s=0;
		for(i=1;i<=maxn;i++)
		{
			if(num[i])
			{
				if(num[i]>1)
					s+=num[i]*(num[i]-1)/2;
				s+=sum[i]*num[i];
			}
		}
		printf("%lld\n",s);
	}		
	return 0;
}