【题解】【算法】- 洛谷 - P3799 妖梦拼木棒（组合数学）

题目背景

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

题目描述

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

答案对 10⁹ + 7 取模。

输入格式

第一行一个整数 n。

第二行 n 个整数，第 i 个整数 a_i 代表第 i 根木棒的长度。

输出格式

一行一个整数代表答案。

输入输出样例

输入 #1复制

4 
1 1 2 2

输出 #1复制

1

说明/提示

数据规模与约定

• 对于30% 的数据，保证 n ≤ 5 × 10³。
• 对于 100% 的数据，保证 1 ≤ n ≤ 10⁵，0 ≤ a_i ≤ 5 × 10³。

思路

暴力即可，详解看代码

AC代码

#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;
const int mod = 1e9 + 7;
const int N