配对（箱）

题目描述

小A举办了一场宴会，请大家吃了蛋糕，并希望大家能交到好朋友。

总共有 n 个人来参加这次宴会，每个人对蛋糕的美味程度都有一个评分，这个评分是一个非负整数。第 i 个人的评分记作 a[i]​。

小A认为如果两个不同的人如果给蛋糕的评分相同的话，他们就能够成功配对。

现在小A想知道，能找到多少组不同的能够成功配对的组合。

【输入格式】

第一行，一个正整数 n。

第二行，包含 n 个用空格分隔的正整数 a[i]​，a[i]​ 表示第 i 个人对蛋糕的评分。

【输出格式】

一行一个非负整数，表示有多少不同的配对组合。

【输入输出样例1】

        输入

    6
    1 2 3 1 1 3

        输出

    4

【输入输出样例2】

        输入

    3
    1 2 3

        输出

    0

【说明提示】

样例 1 解释：有 4 组不同的配对。分别是第 1 个人和第 4 个人，第 1 个人和第 5 个人，第 4 个人                        和第 5 个人，第 3 个人和第 6 个人。

样例 2 解释：不存在合理的配对。

【数据范围】

对于 70% 的数据，1≤n≤103。

对于 100% 的数据，1≤n≤106，每个人对蛋糕的评分都是0∼100之间的整数。

  思路

看到这道题，相信大家第一想到了直接组合，和本人相对一样，这的确是个方法，但是……

#include<bits/stdc++.h>
using namespace std;

const int N = 1e7;
long long n, m, a[N];

int main()
{
	cin >> n;
	for(long long i = 1; i <= n; i++)
	{
		cin >> a[i];
	}
	for(long long i = 1; i <= n; i++)
	{
		for(long long j = i + 1; j <= n; j++)
		{
			if(a[i] == a[j])
			{
				m++;
			}
		}
	}
	cout << m;
	
	return 0;
}

会超时！

所以要用箱的方法去做，每次进行记录，结合等差数列求和公式就行了

我们有时候要学会换一种思想解决问题😊

AC代码

#include<bits/stdc++.h>
using namespace std;

const int N = 1e7;
long long n, m, d[N], maxx = -1;

int main()
{
	cin >> n;
	for(long long i = 1; i <= n; i++)
	{
		long long a;
		cin >> a;
		maxx = max(maxx, a);
		d[a]++;
	}
	for(long long i = 0; i <= maxx; i++)
	{
		m += d[i] * (d[i] - 1) / 2;
	}
	cout << m;
	
	return 0;
}