等边三角形

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

输入格式

首先输入一个整数 $(3\le n\le 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<bits/stdc++.h>
using namespace std;
int n,sum,p[10005]; 
int flag=0;

void dfs(int x,int y,int z,int d){
    if(flag) return ;
	if(x==y&&y==z&&d==n){
		flag=1;
		return ;
	}
	if(x>sum/3||y>sum/3||z>sum/3||d>=n) return ;
	
	dfs(x+p[d],y,z,d+1);
	dfs(x,y+p[d],z,d+1);
	dfs(x,y,z+p[d],d+1);
}

int main(){
	cin>>n;
	for(int i=0;i<n;i++){
		cin>>p[i];
		sum+=p[i];
	}
    if(sum%3){
        cout<<"no"<<endl;
    }
    else{
        dfs(0,0,0,0);
	    if(flag){
		    cout<<"yes"<<endl;
	    }
	    else{
		    cout<<"no"<<endl;
	    }
    }
}