蒜头君手上有一些小木棍,它们长短不一,蒜头君想用这些木棍拼出一个等边三角形,并且每根木棍都要用到。 例如,蒜头君手上有长度为 11,22,33,33 的4根木棍,他可以让长度为11,22 的木棍组成一条边,另外 22 跟分别组成 22 条边,拼成一个边长为 33 的等边三角形。蒜头君希望你提前告诉他能不能拼出来,免得白费功夫。
输入格式
首先输入一个整数 n(3 \le n \le 20)n(3≤n≤20),表示木棍数量,接下来输入 nn 根木棍的长度 p_i(1 \le p_i \le 10000)pi?(1≤pi?≤10000)。
输出格式
如果蒜头君能拼出等边三角形,输出"yes"
,否则输出"no"
。
样例输入1
5 1 2 3 4 5
样例输出1
yes
样例输入2
4 1 1 1 1
样例输出2
no
这道题目又是一道选择的题目,我们有三种选择的方法。
1:我们把这根木棍放到第一条边上。
2:我们把这根木棍放到第二条边上。
3:我们把这根木棍放到第三条边上。
对应三个dfs
。与此同时处理好递归出口的问题就好了。
ac
#include <bits/stdc++.h>using namespace std;int n,l[25],t=0,sum=0;void dfs(int k,int l1,int l2,int l3)//k代表将小标为k的棒放到第几个边 {if(l1>sum||l2>sum||l3>sum)return ;if(l1==sum&&l2==sum&&l3==sum){cout<<"yes";exit(0);return ;}if(k==n)//最后一项已经放置过 回溯 return ;dfs(k+1,l1+l[k],l2,l3);dfs(k+1,l1,l2+l[k],l3);dfs(k+1,l1,l2,l3+l[k]);
}int main(){cin>>n;for(int i=0;i<n;i++){cin>>l[i];sum+=l[i];}if(sum%3)//不是整数 直接排除: {cout<<"no";return 0;}sum/=3;dfs(0,0,0,0); if(!t)cout<<"no";return 0;
}