E1. Close Tuples (easy version)

最新推荐文章于 2022-08-17 15:20:00 发布

_Rikka_

最新推荐文章于 2022-08-17 15:20:00 发布

阅读量444

点赞数

分类专栏：题解算法文章标签： codeforces

本文链接：https://blog.csdn.net/asbbv/article/details/111244227

版权

题解同时被 2 个专栏收录

63 篇文章 0 订阅

订阅专栏

算法

41 篇文章 1 订阅

订阅专栏

博客内容主要介绍了如何利用C++编程解决一个算法问题，即统计数组中每个元素出现的次数，并根据出现次数计算可以形成的特定组合。通过遍历一个映射(map)来记录每个数字的频率，然后根据计数计算可以组成三个相同数字的组合，以及两个相同数字加一个相邻数字的组合。博主特别提到了需要考虑整型溢出问题，以及在计算过程中更新映射以避免重复计数的重要性。

摘要由CSDN通过智能技术生成

题目

原谅兄弟我只A了easy…所以就写个easy的题解吧…

思路：用一个map记录各个值的次数，然后如果某数字数量>=3则可以构成n*(n-1)(n-2)个相同的3个数组成的元组，如果某数字(a)数量>=2则可以构成n(n-1)/2*(a-1的数量+a-2的数量+a+1的数量+a+2的数量)个两个数字为a，其余一个为a+1或a-1或a+2或a-2的元组,如果某数字a的数量>=1则可以构成a的数量a-1的数量a+1的数量三个数字不同（a-1,a,a+1）的元组。对了，还要注意下开long long，因为两个2e5相乘会爆int,我因此还wa了一发。

Code:

#include<iostream>
#include<string>
#include<map>
#include<algorithm>
#include<memory.h>
#include<cmath>
#include<bitset>
#define pii pair<int,int>
#define FAST ios::sync_with_stdio(false),cin.tie(0),cout.tie(0)
using namespace std;
typedef long long ll;
const int Max = 1e6 + 5;
int lst[Max];
int Mod = 1e9 + 7;


int main()
{
	FAST;
	int t;cin >> t;
	while (t--)
	{
		ll n;cin >> n;
		map<ll, ll> ma;
		for (int i = 1;i <= n;i++)
		{
			int g;cin >> g;
			ma[g]++;
		}
		ll sum = 0;
		for (auto i = ma.begin();i != ma.end();i++)
		{
			if (i->second >= 3)
			{
				ll p = i->second;
				sum += (p * (p - 1) * (p - 2) / 6);
			}
			ll p = i->first;
			sum += (ma[p - 1] * ma[p] * ma[p + 1]);
			if (i->second >= 2)
			{
				sum += ((ma[p] * (ma[p] - 1) / 2) * (ma[p - 1] + ma[p + 1]+ma[p+2]+ma[p-2]));
			}
			if (ma[p - 1] == 0)ma.erase(ma.find(p - 1));
			if (ma[p + 1] == 0)ma.erase(ma.find(p + 1));
			if (ma[p + 2] == 0)ma.erase(ma.find(p + 2));
			if (ma[p -2] == 0)ma.erase(ma.find(p -2));
		}
		cout << sum << endl;
	}
}