Sticks POJ - 1011(DFS)
链接:http://poj.org/problem?id=1011
题意:给定 n 根木棍,恰好凑成长度相同的大木棍,问大木棍的最短长度是多少。 ( 1 ≤ n ≤ 64 ) (1\le n \le 64) (1≤n≤64)
思路:搜索的想法有很多种
- dfs(now,len,k)表示当前有k根小木棍要凑成长度为len的大木棍,当前已经凑成的量是now
- dfs(now,len,cnt,total)表示我们需要凑total根长度为len的木棍,现在已经凑成了cnt根,当前拥有的长度为now
剪枝:
- 当 n o w + a [ i ] = = l e n now+a[i]==len now+a[i]==len时,如果此时都不能搜索成功的话,那么这第i根木棍永远都不可能被用到,直接返回false
- 当 n o w = = 0 now==0 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 ≤ n ≤ 64 ) (1\le n \le 64) (1≤n≤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;
}