DFS搜索

Sticks POJ - 1011（DFS）

链接：http://poj.org/problem?id=1011

题意：给定 n 根木棍，恰好凑成长度相同的大木棍，问大木棍的最短长度是多少。$（1\le n\le 64）$

思路：搜索的想法有很多种

1. dfs(now,len,k)表示当前有k根小木棍要凑成长度为len的大木棍，当前已经凑成的量是now
2. dfs(now,len,cnt,total)表示我们需要凑total根长度为len的木棍，现在已经凑成了cnt根，当前拥有的长度为now

剪枝：

1. 当$now+a[i]==len$时，如果此时都不能搜索成功的话，那么这第i根木棍永远都不可能被用到，直接返回false
2. 当$now==0$时，这个时候，通过了上面的搜索但没有成功。拿0去凑，后面依然搜索不到答案，说明这一层的分支就没有答案，返回上一层。

#include <cstdio>
#include <algorithm>
#include <cstring>
#define ll long long
using namespace std;
const int maxn=65;

int n,a[maxn],visit[maxn];

bool dfs(int pos,int len,int now,int cnt)
{
    if(now==0&&cnt==0) return 1;
    if(cnt==0) return 0;
    for(int i=pos; i<=n; ++i)
    {
        if(!visit[i])
        {
            if(a[i]+now<len)
            {
                visit[i]=1;
                if(dfs(i+1,len,a[i]+now,cnt-1)) return 1;
                visit[i]=0;
            }
            if(a[i]+now==len)
            {
                visit[i]=1;
                if(dfs(1,len,0,cnt-1)) return 1;
                visit[i]=0;
            }
            if(a[i]+now==len) return 0;
            if(now==0) return 0;
        }
    }
    return 0;
}
bool cmp(int a,int b)
{
    return a>b
}
int main()
{
    while(scanf("%d",&n)&&n)
    {
        int sum=0;
        for(int i=1; i<=n; ++i)
            scanf("%d",&a[i]),sum+=a[i];
        sort(a+1,a+1+n,cmp);
        for(int i=a[1]; i<=sum; ++i)
        {
            memset(visit,0,sizeof(visit));
            if(sum%i==0&&dfs(1,i,0,n))
            {
                printf("%d\n",i);
                break;
            }
        }
    }
    return 0;
}

Sticks POJ - 1011

链接：http://poj.org/problem?id=1011

题意：给定 n 根木棍，问能否恰好凑成一个正方形（4 根长度相同的木棍）输出 yes or no$（1\le n\le 64）$

思路：同上，比上题更简单

#include <cstdio>
#include <algorithm>
#include <cstring>
#define ll long long
using namespace std;
const int maxn=10000+5;

int t,n,a[maxn],visit[maxn];

bool dfs(int pos,int len,int now,int cnt)
{
    if(now==0&&cnt==0) return 1;
    if(cnt==0) return 0;
    for(int i=pos; i<=n; ++i)
    {
        if(!visit[i])
        {
            if(a[i]+now<len)
            {
                visit[i]=1;
                if(dfs(i+1,len,a[i]+now,cnt-1)) return 1;
                visit[i]=0;
            }
            if(a[i]+now==len)
            {
                visit[i]=1;
                if(dfs(1,len,0,cnt-1)) return 1;
                visit[i]=0;
            }
            if(a[i]+now==len) return 0;
            if(now==0) return 0;
        }
    }
    return 0;
}

int main()
{
    scanf("%d",&t);
    while(t--)
    {
        scanf("%d",&n);
        int sum=0;
        for(int i=1; i<=n; ++i)
            scanf("%d",&a[i]),sum+=a[i];
        if(sum%4!=0) puts("no");
        else
        {
            memset(visit,0,sizeof(visit));
            bool ok=dfs(1,sum/4,0,n);
            puts(ok?"yes":"no");
        }
    }
    return 0;
}