P3799 妖梦拼木棒

最新推荐文章于 2024-11-12 21:59:57 发布

Rachel caramel

最新推荐文章于 2024-11-12 21:59:57 发布

阅读量151

点赞数

分类专栏：洛谷文章标签：算法

本文链接：https://blog.csdn.net/Rachelhello123/article/details/119756262

版权

洛谷专栏收录该内容

44 篇文章 1 订阅

订阅专栏

这篇博客讨论了如何使用算法高效地解决一个火柴棒问题，即如何从给定的一组火柴棒中选择四根形成正三角形。作者首先尝试了简单的排序和遍历方法，但由于其高复杂度导致超时。接着，采用了组合数学的方法，通过存储每个数字的出现次数，减少了计算时间，将时间复杂度降低到O(n²/2)。博客中给出了两种不同思路的代码实现，并最终给出了优化后的解决方案。

摘要由CSDN通过智能技术生成

在这里插入图片描述
思路：
1.既然是选四根火柴组成正三角形，那么一定是a、b、a+b、a+b
2.一开始的想法就是，直接sort排序，然后遍历，如果后面那个数与前面的相同（目的是找出两根相同长度的火柴，即a+b），就往前找a、b，依次进行++。虽然可以过样例，但是实际操作起来很麻烦，很容易算错且复杂度为O（n³），由于题目给定的数据范围所以显然tle

#include <cstdio>
#include <iostream>
#include <algorithm>
using namespace std;
int n,a[11111],ans[111];
int cnt=0;
void search(int m,int l)
{
	for(int i=1;i<m;i++)
	{
		for(int j=i+1;j<m;j++)
		{
			if(a[i]+a[j]==l) cnt++;
		}
	}
}
int main()
{
	cin>>n;
	for(int i=1;i<=n;i++) cin>>a[i];
	sort(a+1,a+n+1);
	for(int i=1;i<=n;i++) 
	{
		if(a[i]==a[i+1])
		{
			search(i,a[i]);
		}
	}
	cout<<cnt<<endl;
	return 0;
}

3.新思路：
从排列组合的角度上来说，存下每个数字的数量，可以节约时间，直接c（m,n）套公式就行，至于a、b的情况在有数组的情况下也会方便得多,可以利用简单的四则运算算出所需要的答案。
如果i的数量大于等于2的话，取的方法次数为也就是c(num[i],2)
如果j=i-j的话判断j的数量是不是大于二，取的方法次数为是c(num[j],2)
如果j!=i-j就判断num[j]和num[i-j]是不是大于1，然后次数为他们的个数即num[j],num[i-j]
时间复杂度为O(n²/2)

#include <cstdio>
#include <iostream>
#include <algorithm>
using namespace std;
long long n,maxx,ans,ans1,ans2;
long long a[111111],num[111111];
const int mod=1e9+7;
int main()
{
	cin>>n;
	for(int i=1;i<=n;i++)
	{
		cin>>a[i];
		num[a[i]]++;
		maxx=max(maxx,a[i]);
	} 
	ans=0;
	for(int i=2;i<=maxx;i++)
	{
		if(num[i]<2) continue;
		else
		{
			ans1=num[i]*(num[i]-1)/2;//cout<<ans<<endl;
			for(int j=1;j<=i/2;j++)
			{
				if(j==i-j&&num[j]>=2)
				{
					ans2=num[j]*(num[j]-1)/2;//cout<<j<<" "<<ans<<endl;
					ans+=(ans1*ans2);//cout<<"asd  "<<ans<<endl;
				}
				if(num[j]>=1&&num[i-j]>=1&&j!=i-j) ans+=ans1*num[j]*num[i-j];
				
			}
		}	
	}
	cout<<ans%mod<<endl;
	return 0;
}