一、题目描述
小 Y 拼木棒
题目背景
上道题中,小 Y 斩了一地的木棒,现在她想要将木棒拼起来。
题目描述
有 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 ll long long
int main(void)
{
int sum;
cin >> sum;
vector<ll>v1(500003, 0);
vector<ll>v2(500003, 0);
for (int i = 0; i < sum; i++)
{
int x; cin >> x;
v1[x]++;
}
//两个不等长的木棍组合长度数目
for (int i = 1; i <= 2500; i++)
{
for (int j = i + 1; j <= 5000 - i; j++)
{
v2[i + j] += v1[i] * v1[j];
}
}
//等长的组合数目
for (int i = 1; i <= 5000; i++)
{
if (v1[i] > 1)
{
v2[2 * i] += v1[i] * (v1[i] - 1) / 2;
}
}
int cnt = 0;
for (int i = 1; i <= 5000; i++)
{
if (v1[i] > 1)
{
cnt += v1[i] * (v1[i] - 1) / 2 * v2[i];
cnt %= 1000000007;
}
}
cout << cnt << endl;
}