P3799 妖梦拼木棒

题目描述

有 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;
}