计蒜客|等边三角形

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

输入格式

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

输出格式

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

样例输入1

51 2 3 4 5

样例输出1

yes

样例输入2

41 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;}