題目背景

上道題中，妖夢斬了一地的木棒，現在她想要將木棒拼起來。

題目描述

有n根木棒，現在從中選4根，想要組成一個正三角形，問有幾種選法？

輸入輸出格式

輸入格式：

第一行一個整數n

第二行n個整數，a1,a2,……an(0<ai<=5000)，代表每根木棒的長度。

輸出格式：

一行一個整數，對1e9+7取模

輸入輸出樣例

4 1 1 2 2

1

說明

對於30%的數據 N<=5000

對於100%的數據 N<=100000

by-szc

題解：枚舉+桶+組合數

代碼：

#include<iostream>
#include
 
<cstdio>
#include<cstring>
const int mod=1e9+7;
using namespace std;
int ans,n,x,mx,sum[100009];
int c(int x){
    return (1LL*x*(x-1)/2)%mod;
}
int main(){
    scanf("%d",&n);
    for(int i=1;i<=n;i++){
        scanf("%d",&x);mx=max(mx,x);sum[x]++;
    }
    for(int i=1;i<=mx;i++){
        
 
for(int j=i;j<=mx;j++){
            if(i+j>mx)break;
            if(i!=j&&sum[i]&&sum[j]&&sum[i+j]>=2)
            ans=(ans%mod+1LL*sum[i]*sum[j]*c(sum[i+j])%mod)%mod;
            else if(i==j)
            ans=(ans%mod+1LL*c(sum[i])*c(sum[i+j])%mod)%mod;
        }
    }
    cout
 
<<ans%mod<<endl;
    return 0;
}

洛谷P3799 妖夢拼木棒