拼三角形

【题目描述】
给你n根木棍，用$4$根木棍能拼出多少个正三角形呢？
【输入格式】
第一行一个整数n
第二行n个整数$Ai$表示第i根木棍的长度。
【输出格式】
正三角形的个数。
答案可能很大，请对$10^9+7$取模。
【样例输入】
4
1 1 2 2
【样例输出】
1
【数据范围】
对于$10%$的数据，满足$1<=n<=10$
对于$50%$的数据，满足$1<=n<=10^3$
对于$100%$的数据，满足$1<=n<=10^6$，且$0<=Ai<=5000$
【分析】
不难想到这4根木棍只能这么分布：x,y,x+y,x+y。
于是枚举$Ai$，用组合数公式计算即可，注意一些细节。

#include<bits/stdc++.h>
using namespace std;
int a[5005]={0};
const int Mod=1e9+7;
int main(){
    int n,x,Max=0,ans=0;
    scanf("%d",&n);
    for (int i=1;i<=n;i++) scanf("%d",&x),a[x]++,Max=max(Max,x);
    for (int i=2;i<=Max;i++){
        int t=a[i]*(a[i]-1)/2;
        for (int j=1;j<=i/2;j++)
            if (j!=i-j) ans=(ans+t*a[j]*a[i-j])%Mod; else ans=(ans+t*a[j]*(a[j]-1)/2)%Mod;
    }
    printf("%d",ans);
}