POJ - 1011 Sticks【DFS】

这道题是一个比较经典的DFS，但是我一开始不会。。。

简单来说，思路是这样的：从小到大遍历所有可能的长度，假设当前遍历到长度为i，那么就要检验这个i是否是满足条件的解，如果是，因为是从小到大遍历，所以是最小的长度，如果不是，则继续遍历。

那么什么才是满足条件的解呢，要求是所有的木棒正好能组合成整数个i长度的木棒。那么怎么判断呢，使用深度优先搜索，对于给定的n个木棒，搜索一种解法能让其组合成整数个i长度的木棒。

其实，这个问题的关键还是看剪枝，主要从三个方面进行剪枝（说实话我做的时候真的怎么都想不到），具体看代码。

 

#include <iostream>
#include <algorithm>
#include <cstdio>
#include <cstring>

using namespace std;

bool cmp(int x, int y) {
	return x > y;
}

int n;
int num[100];
bool book[100];
//剩余木棒数，剩余长度，假定的解的长度，数组开始搜索的下标
//之所以设定一个开始搜索的下标是因为，我们先从大到小排序，如果选定了当前木棒，
//那么比这个木棒大的都不会再选了，因此直接从下一个位置开始就好。
bool dfs( int res , int len , int x , int start) {
	if (!res && !len) return true;
    if (!res) return false;
	if (!len) len = x , start = 1;
	for (int i = start; i <= n; i++) { 
		if (book[i]) continue;
		if (len >= num[i]) {
			book[i] = 1;
			if (dfs(res - 1, len - num[i], x , i)) return true;
			else {
				book[i] = 0;
				if (x == len || len == num[i]) return false; 
				//（1）如果剩余长度等于假定的解的长度，
				//且上边dfs返回false，说明这个假定的解的长度不是真的解的长度，直接跳过。
				//（2）如果剩余长度等于num[i]，且上边dfs返回false，说明这个剩余长度不行，得不到解，跳过
			}
			while (num[i] == num[i + 1]) i++; //（3）如果num[i]不行，那么其他相同长度的也不行，可以跳过
		}
	}
	return false;
}
int main() {
	while (cin >> n, n) {
		int init = 1;
		int sum = 0;
		for (int i = 1; i <= n; i++) {
			scanf("%d", num + i); sum += num[i]; init = max(num[i], init);
		}

		sort(num + 1, num + n + 1, cmp); // 从小到大排序
		for (int i = init; i <= sum; i++) {
			memset(book, 0, sizeof(book));
			if (sum % i) continue;
			if (dfs(n, i, i , 1)) {
				cout << i << endl; break;
			}
		}
	}

	return 0;
}