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

题目背景

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

题目描述

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

答案对 10⁹ + 7 取模。

输入格式

第一行一个整数 n。

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

输出格式

一行一个整数代表答案。

输入输出样例

输入 #1复制

4
1 1 2 2

输出 #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 = 1e5 + 5;
int n, maxj, t, res;
//sum数组存储的是不同长度的木棍的数量，下标是长度，大小是数量
int sum[N];
int main(void)
{cin >> n;for(int i = 0; i < n; i ++){//t相当于中间变量cin >> t;//类似桶排序存入数组中，简化问题sum[t] ++;//找到最长的那个木棍maxj = max(maxj, t);}//i从2开始即可，长度为1的木棍没意义，不可能分成两个木棍for(int i = 2; i <= maxj; i ++){//如果木棍数大于等于2，即可以有两个相等的木棍组成正三角形的两个边if(sum[i] >= 2){//找到另外一边的其中一根木棍，长度大于i / 2也没有什么意义，因为另外一根木棍可以通过i - j求得它的长度for(int j = 1; j <= i / 2; j ++){int k = i - j;//两根木棍长度一样if(k == j)//类似排列组合，第三边两根长度一样的木棍，那么只需要在这种长度的木棍里选两根，组成另外两边的两根一种长度的木棍里选两根就可res += (sum[k] * (sum[k] - 1) / 2 % mod)* (sum[i] * (sum[i] - 1) / 2 % mod);else//类似排列组合，第三边的长度不一样的木棍，其中一种长度的木棍里选一根木棍，另一种长度的木棍里选一根木棍，组成另外两边的两根一种长度的木棍里选两根就可res += (sum[k] % mod) * (sum[j] % mod) * (sum[i] * (sum[i] - 1) / 2 % mod);res %= mod;}}}cout << res << endl;return 0;
}

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

洛谷[P3799 妖梦拼木棒] {暴力枚举} 奋斗的珂珂~
洛谷[P3799 妖梦拼木棒] {暴力枚举} 题目背景上道题中,妖梦斩了一地的木棒,现在她想要将木棒拼起来. 题目描述有 n 根木棒,现在从中选 4 根,想要组成一个正三角形,问有几种选法? 答案 ...
洛谷 P3799 妖梦拼木棒【枚举/组合数学】
题目背景上道题中,妖梦斩了一地的木棒,现在她想要将木棒拼起来. 题目描述有 nnn 根木棒,现在从中选 444 根,想要组成一个正三角形,问有几种选法? 答案对 109+710^9+7109+7 ...
洛谷-P3799 妖梦拼木棒
题目背景上道题中,妖梦斩了一地的木棒,现在她想要将木棒拼起来. 题目描述有 nn 根木棒,现在从中选 44 根,想要组成一个正三角形,问有几种选法? 答案对 10^9+7109+7 取模. 输入格 ...
洛谷P3799 妖梦拼木棒
题目链接题目背景上道题中,妖梦斩了一地的木棒,现在她想要将木棒拼起来. 题目描述有n根木棒,现在从中选4根,想要组成一个正三角形,问有几种选法? 输入格式: 第一行一个整数n 第二行n个整数,a ...
洛谷P3799 妖梦拼木棒题解
//简单的思路:四根必有两根相同,所以咱们就先选两根组合成第三根,确定另外相同的两根的长度 //分为两种情况,一.预选两根相同长度.二.预选两根不同长度. //所有情况是预选*(确定的另外两根) #i ...
洛谷 P3799 妖梦拼木棒
题目这道题主要考组合数学,一开始我没注意数据范围,傻乎乎地模拟,结果复杂度爆炸10个TLE.这道题是算出来的. 思路是用一个栈把相同长度个数超过2的木棍用栈保存下来,同时用一个数组存下所有长度木棍的 ...
P3799 妖梦拼木棒 (组合数学)
题目背景上道题中,妖梦斩了一地的木棒,现在她想要将木棒拼起来. 题目描述有n根木棒,现在从中选4根,想要组成一个正三角形,问有几种选法? 输入输出格式输入格式: 第一行一个整数n 第二行n个整数 ...
[luogu P3799] 妖梦拼木棒
[luogu P3799] 妖梦拼木棒题目描述解决过程思路代码感想题目描述点击此处查看题目描述解决过程暴力不多阐述,很显然是无法AC的解决此题需要用到组合数思路这道题要求拼成等 ...
P3799 妖梦拼木棒——枚举+组合数学
妖梦拼木棒题目背景上道题中,妖梦斩了一地的木棒,现在她想要将木棒拼起来. 题目描述有 nnn 根木棒,现在从中选 444 根,想要组成一个正三角形,问有几种选法? 答案对 109+710^9+7 ...

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

题目背景

题目描述

输入格式

输出格式

输入输出样例

说明/提示

数据规模与约定

思路

AC代码

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

最新文章

热门文章