poj 1011 Sticks (dfs+剪枝)

dfs+剪枝的经典好题

我想到的剪枝方法只有4个，所以代码比较慢，AC用时297ms。

剪枝：

1.在组成新棒的时候，如果拿可用的第一根棒去拼新棒但是失败了，就不用再试了，直接返回上一层，因为我们可以看到，每次     调用dfs(cur+1, compelet, len)，cur是递增的！即如果用cur = k去拼型棒，但是失败了，如果不剪枝，下一步就要用cur = k+1       去拼新棒，然后再递归调用dfs(cur+1 = k+2, ,)  dfs(cur+2 = k+3, ,)。。。。这样k号棒就被空开了，这是不允许的，所以这里         有一个剪枝。

2.显然。

3.如果当前的可用棒和前面的某个棒长度相同，但是前面的那根棒失败了，那就直接return 0。

4.显然。

#include <iostream>
#include <cstdio>
#include <algorithm>
#include <string>
#include <cstring>
using namespace std;
int L, num, a[100], vis[100], n, sum;
 
bool cmp(int _a, int _b)
{
	return _a > _b;
}
 
int dfs(int cur, int compelet, int len)
{
	int i;
	if(L == len)
	{
		compelet ++;
		if(compelet == num)
			return 1;
		else
		{
			for(cur = 0; vis[cur]; cur++)
			;
			vis[cur] = 1;
			if(dfs(cur+1, compelet, a[cur]))// 剪枝1
				return 1;
			vis[cur] = 0;
		}
	}
	else
	{
		for(i = cur; i < n; i++)
		{
			if(!vis[i]&&a[i]<=L-len))// 剪枝2
                        {
                                if(i != 0 & &a[i] == a[i-1] && !vis[i-1]) // 剪枝3
                                   continue;
				vis[i] = 1;
				if(dfs(i+1, compelet, len+a[i]))
					return 1;
				vis[i] = 0;
				
			}
		}
	}
	return 0;
}
		
int main()
{
	//freopen("zztest.txt","r",stdin);
	//freopen("zans.txt","w",stdout);
	int flag = 0;
	int t = 1;
	while(scanf("%d", &n) && n)
	{
		sum = 0;
		for(int i = 0; i < n ;i++)
		{
			scanf("%d", &a[i]);
			sum += a[i];
		}
		sort(a, a+n, cmp);
		memset(vis, 0, sizeof(vis));
		for(L = a[0]; ;L ++)
		{
			if(sum % L)	continue;// 剪枝4
			num = sum / L;
			if(dfs(0, 0, 0))
			{
				printf("%d\n", L);
				break;
			}
			memset(vis, 0, sizeof(vis));
		}
	}
	return 0;
}