等边三角形---dfs

蒜头君手上有一些小木棍，它们长短不一，蒜头君想用这些木棍拼出一个等边三角形，并且每根木棍都要用到。 例如，蒜头君手上有长度为 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)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

这也是典型的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;
}

 

错误的搜索，之前学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;
}