洛谷P3799 妖梦拼木棒

原题链接：传送门

---

前言

这题难度十分简单，很单纯的暴力搜索 + 数学组合。
但是，这题我还是给WA + RE了多次，之所以写下这篇文章也是为了告诫自己，算法路漫漫，一不注意指不定就淹死了。

思路

读完题后看数据范围，发现木棒数量范围在 $[1,1e5]$ ，如果以木棒来进行暴力搜索是必然超时的。
而木棒的长度范围在 $[0,5e3]$ ，显然，对长度进行统计再暴力搜索是合适的。

正解应该是在暴搜中使用到组合数，不过可以化简一下，因为组成的三角形要的是正三角形，即边长相同，其中一条边由两根木棒组成（假设长度为a,b），剩余的两条边的木棒（假设长度为c,d）长度就是已知的。
不难发现，会有两种情况：
① $a=b,c=d=a+b$
② $a!=b,c=d=a+b$

对于①，因为 $a=b$，且 $c=d$，对答案的贡献就是长度为 $a(a=b)$ 的组合数与长度为 $c(c=d)$ 的组合数的组合，即设长度为a的木棒数量有$n_1$个，长度为c的木棒数量有$n_2$个，则贡献为 $C_{n1}^2\ast C_{n2}^2$ 。

对于②，因为 $a!=b$，则由两根木棒的组合数应该为a的数量(设为$n_1$)与b的数量(设为$n_2$)的组合，即 $n_1\ast n_2$ 。而 $c=d$ (设数量为$n_3$)依旧成立，所以贡献为 $n_1\ast n_2\ast C_{n3}^2$ 。

综合以上，统计木棒长度后，进行暴搜 + 组合数，即为答案。

注：此处有可化简的地方，因为涉及组合数的不论是①还是②，都只取2个，可以发现能把组合数公式化成 $\frac{(n-1{)}^2+(n-1)}{2}$ 。

代码

#include <bits/stdc++.h>
#define LL long long

using namespace std;
const int maxn = 5e3;
const LL MOD = 1e9 + 7;
LL vis[maxn + 1];

LL Sn(LL n){ return  (n * n + n) / 2; }

int main()
{
	int n; cin >> n;
	LL ans = 0;
	for (int i = 0; i < n; i++) {
		int x; cin >> x; vis[x]++;
	}

	for (int i = 1; i <= maxn; i++)
		for (int j = i; i + j <= maxn; j++)
			if (i == j && vis[i] >= 2 && vis[i + j] >= 2) ans = (ans + (Sn(vis[i] - 1) * Sn(vis[i + i] - 1))) % MOD;
			else if (i != j && vis[i] > 0 && vis[j] > 0 && vis[i + j] >= 2) ans = (ans + (vis[i] * vis[j] * Sn(vis[i + j] - 1))) % MOD;

	cout << ans;
	return 0;
}

本篇文章仅代表个人观点，如有不同意见欢迎交流。