hdu 4777 USACO ORZ dfs

题目链接：哆啦A梦传送门

题意：给你n条边，用这n条边组成一个三角形，问：有多少中不同的可能性？

题解：

直接暴搜。

参考博客：https://www.cnblogs.com/kuangbin/archive/2012/09/10/2679034.html

#include<cstdio>
#include<algorithm>
#include<cstring>
#include<map>

using namespace std;

typedef long long LL;
int n,tot,ans,num[20];
map<LL,int> ma;

void dfs(int a,int b,int c,int index)
{
    if(a>b) swap(a,b);
    if(a>c) swap(a,c);
    if(b>c) swap(b,c);///从小到大排

    if(2*c>ans) return; ///剪枝
    if(index==n)
    {
        if(a+b>c){
            LL t=a*150000*150000+b*150000+c;
            if(ma[t]==0){
//                    printf("a=%d,b=%d,c=%d\n",a,b,c);
            tot++;
            ma[t]=1;///用map来标记
            }
        }
        return;
    }

    int l1=a+num[index+1];
    int l2=b;
    int l3=c;
    if(l1>l2) swap(l1,l2);
    if(l1>l3) swap(l1,l3);
    if(l2>l3) swap(l2,l3);
    dfs(l1,l2,l3,index+1);

    int t1=a;
    int t2=b+num[index+1];
    int t3=c;
    if(t1>t2) swap(t1,t2);
    if(t1>t3) swap(t1,t3);
    if(t2>t3) swap(t2,t3);
    if(l1!=t1||l2!=t2||l3!=t3)
        dfs(t1,t2,t3,index+1);

    int r1=a;
    int r2=b;
    int r3=c+num[index+1];
    if(r1>r2) swap(r1,r2);
    if(r1>r3) swap(r1,r3);
    if(r2>r3) swap(r2,r3);
    if(l1!=r1||l2!=r2||l3!=r3)
        if(t1!=r1||t2!=r2||t3!=r3)
        dfs(r1,r2,r3,index+1);

    return;

}

int main()
{
    int ncase;

    scanf("%d",&ncase);

    while(ncase--)
    {
        ma.clear();
        scanf("%d",&n);

        ans=0;///存储边长总和
        for(int i=1;i<=n;i++)
        {
            scanf("%d",&num[i]);
            ans+=num[i];
        }
        tot=0;///注意初始化
        dfs(0,0,0,0);
        printf("%d\n",tot);
    }
    return 0;
}

 

代码二：

#include<cstdio>
#include<algorithm>
#include<cstring>
#include<set>
using namespace std;

typedef long long LL;

set<LL> S;
int a[20],tot,ans,n;

void dfs(int x,int y,int index)
{
    int z=ans-x-y;
    if(index==n)
    {
        if(x<=y&&y<=z&&x+y>z)
        {
            S.insert(x*100000+y);///用set来判重

        }
        return;
    }
    dfs(x+a[index+1],y,index+1);
    dfs(x,y+a[index+1],index+1);
    dfs(x,y,index+1);
    return;
}
int main()
{
    int ncase;
    scanf("%d",&ncase);


    while(ncase--)
    {
        S.clear();
        scanf("%d",&n);

        ans=0;
        for(int i=1;i<=n;i++)
            scanf("%d",&a[i]),ans+=a[i];

        dfs(0,0,0);

        printf("%d\n",S.size());
    }
    return 0;
}

 

两种dfs比较：

显然第一种dfs时间复杂度优于第二种。