计蒜客 | | 等边三角形

等边三角形

2000ms  131072K

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

输入格式  

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

输出格式 

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

样例输入1  

5

1 2 3 4 5

样例输出1 

yes

样例输入2  

4

1 1 1 1

样例输出2  

no

 

提示

提前计算出每条边的长度，用 dfs 搜索，如果某条边的长度过大，停止对这个分支的搜索，这一个过程称为剪枝。

#include <iostream>
using namespace std;
int n,flag=0,sum=0;
int length[20];
int group[3]= {0};//设第一条边所在的组为0组 
void dfs(int i,int group_num) {
	group[group_num]+=length[i];
	if(group[1]>sum/3.0||group[0]>sum/3.0||group[2]>sum/3.0)
	{
		group[group_num]-=length[i];
		return ;
	}
	
	if(group[0]==sum/3.0&&group[1]==sum/3.0)
	{	flag=1;//符合 
		return ;
	}
	if(i+1<n) {
		if(!flag)dfs(i+1,0);
		if(!flag)dfs(i+1,1);
		if(!flag)dfs(i+1,2);
	} 
	group[group_num]-=length[i];
	return ;
 
}
 
int main() {
	cin>>n;
	for(int i=0; i<n; i++) {
		cin>>length[i];
		sum+=length[i];
	}
	dfs(0,0);
	if(flag)cout<<"yes";
	else cout<<"no";
	return 0;
}