洛谷P3799 妖梦拼木棒题解

原创已于 2022-09-19 17:00:34 修改 · 528 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#c++ #算法

于 2022-05-04 22:11:47 首次发布

洛谷刷题专栏收录该内容

16 篇文章

订阅专栏

该博客探讨了一道数学问题，涉及四根木棍如何组合使得两根木棍之和等于另外两根。通过预选相同或不同长度的木棍，计算所有可能的组合，最终求出总方案数。算法实现包括对每种长度木棍的计数，并考虑组合的条件进行求解。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

//简单的思路：四根必有两根相同，所以咱们就先选两根组合成第三根，确定另外相同的两根的长度
//分为两种情况，一、预选两根相同长度。二、预选两根不同长度。
//所有情况是预选*（确定的另外两根）
#include<iostream>
using namespace std;
long long int n,arr[5500],temp,sum;
int maxn;
void shuru()
{
	cin >> n;
	for (int i = 1; i <= n; i++)
	{
		cin >> temp;
		arr[temp]++;//存储长度相同的根数
		if (temp > maxn)
			maxn = temp;
	}
}
int main()
{
	freopen("title.in", "r", stdin);
	shuru();
	for (int i = 1; i <=maxn; i++)
	{
		for (int d = i; d <=maxn; d++)
		{
			if (arr[i]>=1 && arr[d]>=1 && i + d <= maxn && arr[i + d] >= 2)
			{
				if (i==d&&arr[i]>=2)
					sum = sum + arr[i + d]*(arr[i+d]-1) * arr[i]*(arr[i]-1)/4;//预选和确定两根都要在总根数中取两根，所以就是取组合的n取2
				else if(i!=d)
				 sum = sum + arr[i+d]*(arr[i + d] - 1) * arr[i] * arr[d]/2;//不相同则不需要n取2，此处n为每个长度的根数
			}	
		}
	}
	cout << sum%(int)(1e9+7);
	return 0;
}