取木棒凑正三角形

有 nn根木棒，现在从中选 44 根，想要组成一个正三角形，问有几种选法？
答案对 10^9+7 取模。
输入格式
第一行一个整数 n。

第二行 n 个整数，第 i 个整数 a i 代表第 i 根木棒的长度。

输出格式
一行一个整数代表答案。

输入输出样例
输入
4
1 1 2 2
输出
1
思路：取四个构成正三角形，则必有a=b=c+d我们取两个之后（即两层循环）便可以求出，因为数据可能很大，而木棒长度并不大，这里我们用桶存储，即把每个不同长度的木棒放在各个数组空间里,这里我们用p数组表示，如果出现一个则使对应的数量+1。
这里我们需要判断当a=b和a!=b时的情况(cc(int n)指的是从n个数中取两个的函数)
当a！=b时，这样的组合的数量=长度为aa的木棒数量×长度为bb的木棒数量×长度为(a+b)(a+b)的数量。
当a==b时，
由于这里已经取走了一根长度为aa的木棒，那么再取一根长度为a的木棒的方案数就要减1。
所以这样的组合的数量=长度为aa的木棒数量×(长度为bb的木棒数量 -1)×长度为(a+b)(a+b)的数量。
下来看代码

#include <iostream>
using namespace std;
#include <algorithm>
#include <math.h>
#include <stdio.h>
#include <string>
#include <stdlib.h>
#include <iomanip>
const int Mod=1e9+7;
int a[100001],book[100001];//桶存储木棒长度 
int cc(int n) //所有组合中取两种  //高中组合公式化简 
{   
   return n*(n-1)/2;
}
int main()
{   int i,j,n,max1=-1;
    long long count1=0;
    cin>>n;
    for(i=0;i<n;i++)
     { cin>>a[i];
       max1=max(a[i],max1);
       book[a[i]]++;
	 }
	 for(i=1;i<=max1;i++)
	   for(j=i;j<=max1;j++)
	    { 
	       if(i==j) //当取的第一个和第二个 相等时 
	        {  count1=(count1+cc(book[i])*cc(book[i+j]))%Mod;
			}
			else
			  { count1=(count1+(book[i]*book[j])%Mod*cc(book[i+j]))%Mod;
			  }
	    
	    
		}
    printf("%d",count1%Mod);
    return 0;
}