Bootstrap

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

题目背景

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

题目描述

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

答案对 109 + 7 取模。

输入格式

第一行一个整数 n。

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

输出格式

一行一个整数代表答案。

输入输出样例

输入 #1复制

4 
1 1 2 2

输出 #1复制

1

说明/提示

数据规模与约定
  • 对于30% 的数据,保证 n ≤ 5 × 103
  • 对于 100% 的数据,保证 1 ≤ n ≤ 105,0 ≤ ai ≤ 5 × 103

思路

暴力即可,详解看代码

AC代码

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