等边三角形

题目:
蒜头君手上有一些小木棍，它们长短不一，蒜头君想用这些木棍拼出一个等边三角形，并且每根木棍都要用到。 例如，蒜头君手上有长度为 11，22，33，33 的4根木棍，他可以让长度为11，22 的木棍组成一条边，另外 22 跟分别组成 22 条边，拼成一个边长为 33 的等边三角形。蒜头君希望你提前告诉他能不能拼出来，免得白费功夫。

输入格式
首先输入一个整数 n(3 \le n \le 20)n(3≤n≤20)，表示木棍数量，接下来输入 nn 根木棍的长度 p_i(1 \le p_i \le 10000)pi​(1≤pi​≤10000)。

输出格式
如果蒜头君能拼出等边三角形，输出"yes"，否则输出"no"。

样例输入1
5
1 2 3 4 5
样例输出1
yes
样例输入2
4
1 1 1 1
样例输出2
no

代码如下：

#include<bits/stdc++.h>
using namespace std;
int n,a[15],sum = 0;
bool f,vis[15];
void dfs(int s,int sum2)//s代表目前拼出的边数，sum2代表这一轮边总和 
{
	if(sum2 == sum / 3) {dfs(s + 1,0);return;} //成功拼出一条，接下去拼下一条 
	if(s == 3) {f = true;return;}//三条已经拼出 
	for(int i = 0;i < n;i++){
		if(!vis[i]){
			vis[i] = 1;
			dfs(s,sum2 + a[i]);
			vis[i] = 0;
		}
	}
}
int main()
{
	//freopen("triangle.in","r",stdin);
	//freopen("triangle.out","w",stdout);
	cin >> n;
	for(int i = 0;i < n;i++){
		cin >> a[i];
		sum += a[i];
	}
	if(sum % 3) cout << "no" << endl;
	else {
		dfs(0,0);
		if(f) cout << "yes" << endl;
		else cout << "no" << endl;
	}
	return 0;	
} 

想要把这些边拼出等边三角形而且每一条边都要用到且只用一次，那么这个等边三角形的边长就是所有边之和除3.那么我们首先要去判断所有边之和是否为3的倍数，如果不是直接no就可以了，如果是3的倍数，再进行dfs，dfs两个参数分别代表目前拼出的边数和这一轮边总和，具体过程上面代码有注释，这里就不多说了。另外注意要回溯