P3799 小 Y 拼木棒

题目背景

上道题中，小 Y 斩了一地的木棒，现在她想要将木棒拼起来。

题目描述

有 $n$ 根木棒，现在从中选 $4$ 根，想要组成一个正三角形，问有几种选法？

答案对 $10^9+7$ 取模。

输入格式

第一行一个整数 $n$。

第二行往下 $n$ 行，每行 $1$ 个整数，第 $i$ 个整数 $a_i$ 代表第 $i$ 根木棒的长度。

输出格式

一行一个整数代表答案。

样例 #1

样例输入 #1

4 
1
1
2
2

样例输出 #1

1

提示

数据规模与约定

• 对于 $30\%$ 的数据，保证 $n\le 5\times 10^3$。
• 对于 $100\%$ 的数据，保证 $1\le n\le 10^5$，$1\le a_i\le 5\times 10^3$。

关于标题：因为一些不可抗力的原因，名称进行了更改。深表歉意。

思路

参考大佬题解，清晰明了。

代码

那个t2没有在循环中赋为0卡了半天。。。

#include<iostream>
#include<algorithm>
#include<cstring>
#include<vector>
#define endl '\n'
using namespace std;

typedef long long LL;// long long 防止中间结果溢出

const int N = 100010;
const int mod = 1e9 + 7;

int n, maxa;// maxa表示所有木棒的最大长度，便于之后枚举
int a[N], nums[5010];// nums表示每种长度木棒的根数
LL ans;

// 求组合数C(x,1)，C(x,2)，x最大5000，结果不会爆int
int C(int x, int y) {

	if (y == 1) {
		return x;
	}
	else if (y == 2) {
		return x * (x - 1) / 2;
	}
}

int main() {
	ios::sync_with_stdio(false), cin.tie(nullptr), cout.tie(nullptr);
	cin >> n;
	for (int i = 1; i <= n; i++) {
		cin >> a[i];
		maxa = max(maxa, a[i]);
		nums[a[i]]++;
	}

	// 枚举长度相等的那两条边的所有长度
	for (int i = 2; i <= maxa; i++) {

		if (nums[i] >= 2) {
			
			int t1 = C(nums[i], 2);// 从这个所有这个长度 i 的木棒中选两根的方案数
			int t2 = 0;// 其余两根长度之和为 i 的方案数

			for (int j = 1; j <= i / 2; j++) {

				t2 = 0;// ！！！！！！！别漏了
				// 要不然如果不存在其余两根长度之和为 i 的方案，那么t2就是上次循环所算出的，就会出错。。。

				// 其余两根长度不等
				if (j != i - j && nums[j] >= 1 && nums[i - j] >= 1) {
					t2 = (LL)C(nums[j], 1) * C(nums[i - j], 1) % mod;
				}

				// 其余两根长度相等
				if (j == i - j && nums[j] >= 2) {
					t2 = (LL)C(nums[j], 2) % mod;
				}

				ans += ((LL)t1 * t2 % mod);
				ans = ans % mod;// 这个别忘了，原来的ans加上新的方案数可能会超出mod
			}
		}
	}

	cout << ans << endl;
	return 0;
}