一道水题 (还是做了一个小时,我太菜了
基本思路:
题里面说,4根棍子拼成一个正三角形(等边三角形)
若设这四根棍子长度为(a,b,c,d)且(a≥b>c≥d)
那很容易得到 (真的很容易):
(a=b=c+d)
自然,我们只需枚举(a)的长度和(c)的长度就行了
长度最长才5k,木棍用桶按长度装一下,遍历的时候也方便。
代码要好好看完啊kora
#include <bits/stdc++.h>
#define c1(a) (a)//带参宏定义,本题不必写函数
#define c2(a) ((a)*((a)-1)/2)
using namespace std;
const int mode=1000000007;
int aaa[5010];//桶
int main()
{
ios::sync_with_stdio(0);
cin.tie(0);
cout.tie(0);//io流解绑
int n;
cin>>n;
for(int i=1;i<=n;i++)
{
int k;
cin>>k;
aaa[k]++;//往桶里扔
}
int ans=0;
for(int a=2;a<=5000;a++)
{
for(int c=1;c<=a/2;c++)//c最大只能到a的一半
{
int d=a-c;//把d算出来判定相等
if(d!=c&&aaa[a]>=2&&aaa[c]>=1&&aaa[d]>=1)//c d不同
{
ans+=((c2(aaa[a])%mode)*(c1(aaa[c])%mode)*(c1(aaa[d])%mode))%mode;
}
else if(d==c&&aaa[a]>=2&&aaa[c]>=2)//c d相同
{
ans+=((c2(aaa[a])%mode)*(c2(aaa[c])%mode))%mode;
}
ans%=mode;
}
}
cout<<ans%mode;//多模几次总没问题的
return 0;
}