妖梦拼木棒
题目背景
上道题中,妖梦斩了一地的木棒,现在她想要将木棒拼起来。
题目描述
有 n n n 根木棒,现在从中选 4 4 4 根,想要组成一个正三角形,问有几种选法?
答案对 1 0 9 + 7 10^9+7 109+7 取模。
输入格式
第一行一个整数 n n n。
第二行往下 n n n 行,每行 1 1 1 个整数,第 i i i 个整数 a i a_i ai 代表第 i i i 根木棒的长度。
输出格式
一行一个整数代表答案。
样例 #1
样例输入 #1
4
1
1
2
2
样例输出 #1
1
提示
数据规模与约定
- 对于 30 % 30\% 30% 的数据,保证 n ≤ 5 × 1 0 3 n \le 5 \times 10^3 n≤5×103。
- 对于
100
%
100\%
100% 的数据,保证
1
≤
n
≤
1
0
5
1 \leq n \le 10^5
1≤n≤105,
1
≤
a
i
≤
5
×
1
0
3
1 \le a_i \le 5 \times 10^3
1≤ai≤5×103。
小丑!
#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;
}