题目描述
有 nn根木棒,现在从中选 4 根,想要组成一个正三角形,问有几种选法?答案对 10^9+7取模。
输入输出格式
输入:第一行一个整数 n。第二行 n 个整数,第 i 个整数 a_i,代表第 i 根木棒的长度。
输出:一行一个整数代表答案。
题目解答
主要是对长度进行计算和处理…因为ai的数值偏小。
注意:
- 两个小棒的长度是否相等
- 注意数组的下标是否越界,和为了效率的控制跳跃
using namespace std;
#include <iostream>
#include <cstring>
#include <cmath>
int a[100005];
int mi=1e9,ma=-1e9;
int main ()
{
int n,i,j;
cin>>n;
int k[n];
for(i=0;i<n;i++)
{
cin>>k[i];
mi=min(mi,k[i]);
ma=max(ma,k[i]);
a[k[i]]+=1;
}
int sum=0;
for(i=mi+1;i<=ma;i++)
{
if(a[i]<=1){continue;} //没有可选
for(j=mi;j<=i/2;j++) //注意范围
{
if(a[j]&&a[i-j]) //有可选的
{
if(j==i-j&&a[j]>=2){ //有两个可选
sum=sum+a[i]*(a[i]-1)/2*a[j]*(a[j]-1)/2;
//cout<<i<<" "<<j<<" "<<(i-j)<<" "<<a[i]*(a[i]-1)/2*a[j]*(a[j]-1)/2<<" "<<sum<<endl;
}
if(j!=i-j)
{
sum=sum+a[i]*(a[i]-1)/2*a[j]*a[i-j];
//cout<<i<<" "<<j<<" "<<(i-j)<<" "<<a[i]*(a[i]-1)/2*a[j]*a[i-j]<<" "<<sum<<endl;
}
sum=sum%1000000007;
}
}
}
cout<<sum;
return 0;
}