ABC200 C - Ringo‘s Favorite Numbers 2 思维+map计数

使用映射统计数组中元素出现次数并计算组合数

原创于 2022-03-21 19:28:10 发布 · 372 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#c++

我的ACM之路同时被 2 个专栏收录

159 篇文章

订阅专栏

好题

22 篇文章

订阅专栏

这段代码主要解决了一个数学问题，通过遍历输入数组，统计每个元素对200取余后的值出现的次数，并存储在映射中。然后计算所有相同余数值的元素之间两两组合的数量，避免了中间结果导致的整数溢出。最后输出组合总数。

原题
 参考：live4m大佬的题解

注意中途可能会爆int。
能组成<a,b>使得b-a是200的倍数的，都满足a%200=b%200；

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
#define INF 0x3f3f3f3f
#define fir(i,a,n) for(int i=a;i<=n;i++)
#define se second
const int N=1e5+10;
int n;
map<ll,ll>mp;
int main()
{
	cin>>n;
	fir(i,1,n) 
	{
		int t;cin>>t;
		mp[t%200]++;
	}
	ll ans=0;
	for(auto u:mp)
	{
		ans+=((ll)u.se*(u.se-1)/2);//这里可能会爆int 
	}
	cout<<ans;
	return 0;
}