洛谷题集——妖梦拼木棒（数组与map运行效率）

在STL库中，我们能够使用map容器自动建立Key－value的对应。key 和 value可以是任意你需要的类型。
假如我们将map作为数组来使用，我们会发现它们的运行效率截然不同！

题目：妖梦拼木棒 洛谷链接
有 n 根木棒，现在从中选 4 根，想要组成一个正三角形，问有几种选法？
答案对10^9+7取模。

输入格式
第一行一个整数 n。
第二行 n 个整数，第 i 个整数 a i
​代表第 i 根木棒的长度。
输出格式
一行一个整数代表答案。

输入输出样例
输入 #1
4
1 1 2 2
输出 #1
1

说明/提示
数据规模与约定
对于 30% 的数据，保证 n≤5×10³。
对于 100% 的数据，保证 1≤n≤10 ⁵，0≤ai ≤5×10 ³。

---

解题思路：
用m数组或m映射存储不同长度木棍的数量。
①、找两根长度为k长棍
②、找两根短棍长度分别为i，j。
保证：k = i + j
③、统计长度为k的正三角形的数量。
当短棍不相等时
ans = ans+m[k] * (m[k]-1) * m[j] * m[i]/2
当短棍长度相等时
ans = ans + m[k] * (m[k]-1) * m[i] * (m[i]-1) / 4
ps:记得取模mod=1e9+7

用数组存储不同长度木棍的数量

#include<bits/stdc++.h>
#define ll long long
using namespace std;
const ll mod = 1e9+7;
ll ans = 0;
int m[500010]