版权声明:本文为博主原创文章,未经博主允许不得转载。 https://blog.csdn.net/z914022466/article/details/70336601
蒜头君手上有一些小木棍,它们长短不一,蒜头君想用这些木棍拼出一个等边三角形,并且每根木棍都要用到。 例如,蒜头君手上有长度为 1,2,3,3 的4根木棍,他可以让长度为1,2 的木棍组成一条边,另外 2 跟分别组成 2条边,拼成一个边长为 3 的等边三角形。蒜头君希望你提前告诉他能不能拼出来,免得白费功夫。
输入格式
首先输入一个整数 n(3≤n≤20),表示木棍数量,接下来输入 n 根木棍的长度 pi(1≤pi≤10000)。
输出格式
如果蒜头君能拼出等边三角形,输出"yes"
,否则输出"no"
。
样例输入1
5 1 2 3 4 5
样例输出1
yes
样例输入2
4 1 1 1 1
样例输出2
no
#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;
}