牛客7 H-xay loves count

最新推荐文章于 2021-09-03 16:17:06 发布

【杰杰】

最新推荐文章于 2021-09-03 16:17:06 发布

阅读量125

点赞数

分类专栏：数学

本文链接：https://blog.csdn.net/p15008340649/article/details/119493981

版权

算法 C++ 数组处理元素乘积组合计算

关键词由CSDN通过智能技术生成

数学专栏收录该内容

47 篇文章 2 订阅

订阅专栏

传送门

思路：

去重排序并记录每个数的个数，对于i和j对应的相同的数a， $a n s + = a 的数量 * a 的数量 * （ a * a ）的数量$ ，对于i和j不相同的数，将上述值*2即可。

#include<bits/stdc++.h>
using namespace std;
#define ll long long
#define endl '\n'
ll vis[1000010];
ll b[1000010];

int main()
{
	int n,cnt = 0;
	cin>>n;
	for(int i = 1; i <= n; i++)
	{
		int a;
		cin>>a;
		if(!vis[a])b[++cnt] = a;
		vis[a]++;
	}
	sort(b+1,b+1+cnt);
	ll ans = 0;
	for(int i = 1; i <= cnt; i++)
	{
		for(int j = i; j <= cnt; j++)
		{
			if(b[i]*b[j] <= 1e6)
			{
				if(b[i]==b[j])
				ans += vis[b[i]]*vis[b[j]]*vis[b[i]*b[j]];
				else
				ans += vis[b[i]]*vis[b[j]]*vis[b[i]*b[j]]*2;
			}
			else break;
		}
	}
	cout<<ans<<endl;
}