C++ 小 Y 拼木棒

---

一、题目描述

小 Y 拼木棒

题目背景

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

题目描述

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

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

输入格式

第一行一个整数 $n$。

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

输出格式

一行一个整数代表答案。

样例 #1

样例输入 #1

4 
1
1
2
2

样例输出 #1

1

提示

数据规模与约定

• 对于 $30\%$ 的数据，保证 $n\le 5\times 10^3$。
• 对于 $100\%$ 的数据，保证 $1\le n\le 10^5$，$1\le a_i\le 5\times 10^3$。

关于标题：因为一些不可抗力的原因，名称进行了更改。深表歉意。

---

二、参考代码

#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;
}

---