妖梦拼木棒（组合数学）

#include
#include
#include
#include
using namespace std;
typedef long long ll;
const int maxn=1e5+10;
const int mod=1e9+7;
ll c[5010][5010],n,a[maxn],num[5010],ans;
int main(){
 scanf("%lld",&n);
 c[0][0]=1;
 for(int i=1;i<=n;i++){
  c[i][0]=c[i][i]=1;
  for(int j=1;j    c[i][j]=(c[i-1][j]+c[i-1][j-1])%mod;
  }
 }
 for(int i=1;i<=n;i++){
  scanf("%lld",&a[i]); num[a[i]]++;
 } 
 for(int i=1;i<=5000;i++){
  for(int j=2*i;j<=5000;j++){
   if(j-i==i){
    if(num[j]>=2&&num[i]>=2){
     ans=ans+c[num[i]][2]%mod*c[num[j]][2]%mod;
     ans%=mod;
    }
   }
   else if(num[j]>=2){
    ans=ans+c[num[i]][1]%mod*c[num[j]][2]%mod*c[num[j-i]][1]%mod;
    ans%=mod;
   }
  }
 }
 cout<  return 0;
}