题目描述

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

首先输入一个整数 n(3≤n≤20)，表示木棍数量，接下来输入 nn 根木棍的长度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

思路：从未被选中中选择一条边，

#include <iostream>
using namespace std;
int p[50];
int n,sum=0;
bool f;     //表示找没找到
bool vis[50];
//cnt木棍的个数，s和，st保证从i元素之后找，不重复
void dfs(int cnt,int s){
    if(f){  
        return;
    }
    if(cnt==3){
        f=true;
        return;
    }
    //当木棍的长度到了所需要的长度的时候开始处理
    if(s==sum/3){
        dfs(cnt+1,0);
    }
    for(int i=0;i<n;i++){
        if(!vis[i]){
            vis[i]=true;
            dfs(cnt,s+p[i]);
            vis[i]=false;
        }
    }
}
int main()
{
    cin>>n;
    for(int i=0;i<n;i++){
        cin>>p[i];
        sum+=p[i];
    }
    if(sum%3!=0){
        cout<<"no"<<endl;
    }else{
        dfs(0,0);
        if(f){
            cout<<"yes"<<endl;
        }
        else{
            cout<<"no"<<endl;
        }
    }
} // namespace std;