洛谷 P3799妖梦拼木棒

最新推荐文章于 2024-06-15 23:36:03 发布

銮銮大魔王

最新推荐文章于 2024-06-15 23:36:03 发布

阅读量1.9k

点赞数 35

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

本文链接：https://blog.csdn.net/m0_65414734/article/details/136752239

版权

洛谷专栏收录该内容

33 篇文章 0 订阅

订阅专栏

该文章展示了如何用C++编程解决一个关于选择木棒组成正三角形的问题，涉及到组合数学和模运算技巧。

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

妖梦拼木棒

题目背景

上道题中，妖梦斩了一地的木棒，现在她想要将木棒拼起来。

题目描述

有 $n$ 根木棒，现在从中选 $4$ 根，想要组成一个正三角形，问有几种选法？

答案对 $10^9+7$ 取模。

输入格式

第一行一个整数 $n$ 。

第二行往下 $n$ 行，每行 $1$ 个整数，第 $i$ 个整数 $a_i$ 代表第 $i$ 根木棒的长度。

输出格式

一行一个整数代表答案。

样例 #1

样例输入 #1

样例输出 #1

提示

数据规模与约定

对于 $30\%$ 的数据，保证 $\le 5 \times 10^3$ 。
对于 $100\%$ 的数据，保证 $\leq n \le 10^5$ ， $\le a_i \le 5 \times 10^3$ 。
小丑！

#include <bits/stdc++.h>
using namespace std;
#define MAX 1000000007
#define ll long long  
map<int , int> mp;
ll C(ll x, ll k) { 
  //求得从n个数中取出k个数的组合
  //此处k=1 / 2，用了特判写法。
  //k = 1 时，C(x, 1) = x;
  //k = 2 时，C(x, 2) = x * (x - 1) / 2;
  return (k == 1ll ? x : x * (x - 1ll) / 2ll) % MAX;
}
int main()
{
	ios::sync_with_stdio(false),cin.tie(0),cout.tie(0);//规模较大 
	int n;
	cin>>n;
	for(int i=0;i<n;i++)
	{
		int bb;
		cin>>bb;
		if(mp.find(bb) == mp.end()) 
	{
		mp[bb] = 1;
	}
	else
	{
		mp[bb]++;
	}
	}
	
	map<int , int> ::iterator it = mp.begin();
	int ans=0;
	for(;it!=mp.end();it++)
	{
		ans%=MAX;
		//注意分情况，奇偶数 
		int x = C(it->second,2); 
		for(int i=1;i<=(it->first)/2;i++)//长度访问一半即可
		//如果偶数特盘在外面，这里仍然应该<=，比如5，就取不到2 
		{
			
//			temp+=((mp[i]%MAX)*(mp[it->first-i]%MAX))%MAX;//超级保险
			//不存在就返回0，不用担心返回值错误
			//对于题意理解错误！不能重复选！！->对于同一种情况下讨论 
			
			int l1 =  mp[i],l2 = mp[it->first-i];//就算在2211的情况下也是对1每次去除两根，完全正确 
//			while(l1!=0&&l2!=0&&x!=0)//这里可以优化 
//			{
//				l1--;l2--;x-=2;ans++;
//			}
			
//			ans+=l1*l2*x/2%MAX ;
			//应该是对x取组合 
			ll j =  it->first-i;
			if (j != i && l2 >= 1 && l1 >= 1) //用来合成的木棒长度不等 
          ans += x * C(l2, 1) * C(l1, 1) % MAX;
        if (j == i && l1 >= 2) //用来合成的木棒长度相等 
          ans += x * C(l2, 2) % MAX;
			
		 }
//		//如果担心乘法过程中越界就改为循环加取余 与上面一起错 
//		ans+=(temp*it->second)%MAX;
//		 //注意奇偶

//		 if(mp[it->first/2]>=2)//注意>=2 
//		 {
//		 	if(it->first%2==0) ans+=C(mp[it->first/2],2)*C(x,2);
//		 }
//			
//			ans%=MAX;
	}
	cout<<ans%MAX; 
	return 0;
}