妖梦拼木棒

妖梦拼木棒

题目背景

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

题目描述

有 $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$。

#include <stdio.h>
#include <stdlib.h>

int main(){
	int a[5005] = {0}; 			//存放对应木棒长度的个数
	int ans = 0, temp;		//临时变量 temp 存储结果ans 
	int n;
	scanf("%d", &n);
	
	int i, j;
	for (i=1; i<=n; i++){
		scanf("%d", &temp);		//读取数据到临时变量 temp存储木棒长度
		a[temp]++;			
	}
	
	for(i=1; i<=2500; i++){
		for(j=i; j<=5000-i; j++){
			int k1 = a[i], k2 = a[j], k3 = a[i+j]; 			//k1 存放第一根木棒 k2存放第二根木棒 k3第三根木棒 
			if(k3 >= 2){									//第三根木棒长度相等 所以要大于两根 并从中选两根 
				if(i == j && a[i] >= 2){
					temp = k1 * (k1-1) / 2 * k3 * (k3 -1) / 2;  //第一根木棒和第二根木棒相等 所以要大于两根 从该长度的木棒中中选两根 
					ans+=temp;
				}
				if(i != j && a[i] >= 1 && a[j] >= 1){
					temp = k1 * k2 * k3 * (k3 -1) / 2;			// 第一根长度和第二根不等 所以都要要大于一根
					ans+=temp;
				}
			}
		}
	}
	printf("%d", ans);

	return 0;
}