首页 > 代码库 > 等边三角形---dfs

等边三角形---dfs

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

输入格式

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

输出格式

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

样例输入1

51 2 3 4 5

样例输出1

yes

样例输入2

41 1 1 1

样例输出2

no


这也是典型的dfs:
对每一根木棍,要么加到a边,要么加到b边,要么加到c边。(因为所有的木棍都要使用)。dfs搜索每一个木棍即可。

 

技术分享
#include<bits/stdc++.h>using namespace std;int n;int l[25];int len;bool dfs(int a,int b,int c,int id){    if(a==b&&b==c&&id==n) return true;    if(a>len||b>len||c>len) return false;    if(id>=n) return false;    if(dfs(a+l[id],b,c,id+1)||dfs(a,b+l[id],c,id+1)||dfs(a,b,c+l[id],id+1)) return true;    return false;}int main(){    cin>>n;    int sum=0;    int maxn=-1;    for(int i=0; i<n; i++)    {        cin>>l[i];        sum+=l[i];        maxn=max(maxn,l[i]);    }    if((sum/3*3)!=sum||maxn>(sum/3))    {        cout<<"no"<<endl;        return 0;    }    int a,b,c;    a=b=c=0;    len=sum/3;    if(dfs(a,b,c,0)) cout<<"yes"<<endl;    else cout<<"no"<<endl;    return 0;}
View Code

 

错误的搜索,之前学dfs的时候,习惯用循环,导致喜欢套用这个模板。。。。。。。。

技术分享
#include<bits/stdc++.h>using namespace std;int n;int a[25];int len;bool vis[25];bool dfs(int sum,int index){    if(sum==len) return true;    if(sum>len) return false;    if(index>=n) return false;    if(dfs(sum+a[index],index+1))    {        vis[index]=1;        return true;    }    else    {        if(dfs(sum,index+1)) return true;    }    return false;}int main(){    cin>>n;    int sum=0;    int maxn=-1;    for(int i=0; i<n; i++)    {        cin>>a[i];        sum+=a[i];        maxn=max(maxn,a[i]);    }    if((sum/3*3)!=sum||maxn>(sum/3))    {        cout<<"no"<<endl;        return 0;    }    len=sum/3;    bool flag=false;    memset(vis,0,sizeof(vis));    for(int i=0; i<n; i++)    {        if(!vis[i])        {            flag=true;            if(!dfs(0,i))            {                cout<<i<<endl;                for(int i=0; i<n; i++)                    cout<<vis[i]<<" ";                cout<<"no"<<endl;                return 0;            }            flag=false;        }    }    if(!flag) cout<<"yes"<<endl;    else cout<<"no"<<endl;}
View Code

 

等边三角形---dfs