poj 1011 木棒长度

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

搜索的经典题目。有一堆不同长度木棒，问你用这些木棒首尾相接所组成的一系列长度相等的木棒的最短长度。

思路就是用DFS搜索，关键在剪枝：

1.答案的长度必然可以整除总长度

2.每次选取的第一根木棒必须能够组成一根目标长度，如果这根不行，那么后续的也不需要做了，直接换下一个长度就可以了

#include<iostream>
#include<stdio.h>
#include<cstring>
using namespace std;
#define N 101
int arr[N];
int n;
int length;
int vis[N];


void qs(int left,int right) {
	if(left>=right) {
		return;
	}
	int tmp=arr[left],i=left,j=right;
	while(i<j) {
		while(j>i&&arr[j]<=tmp) {
			j--;
		}
		while(j>i&&arr[i]>=tmp) {
			i++;
		}
		if(i<j)
		swap(arr[i],arr[j]);
	}
	swap(arr[left],arr[j]);
	qs(left,j-1);
	qs(j+1,right);

}

int dfs(int now_len,int k,int cnt,int need) {
	if(cnt==need) {
		return 1;
	}
	if(now_len==length) {
		return dfs(0,0,cnt+1,need);
	}
	int i,pre=-1;
	for(i=k; i<n; i++) {
		if(vis[i]==0&&now_len+arr[i]<=length&&pre!=arr[i]) {
			vis[i]=1;
			pre=arr[i];
			if(dfs(now_len+arr[i],i+1,cnt,need)) {
				break;
			}
			vis[i]=0;
			if(k==0){
				return 0;
			}
		}
	}
	if(i==n) {
		return 0;
	} else {
		return 1;
	}


}

int main() {
	while(~scanf("%d",&n)&&n!=0) {
		int sum=0;
		for(int i=0; i<n; i++) {
			scanf("%d",&arr[i]);
			sum+=arr[i];
		}
		qs(0,n-1);
		for(length=arr[0]; length<=sum/2; length++ ) {
			if(sum%length==0) {
				memset(vis,0,sizeof(vis));
				if(dfs(0,0,0,sum/length)) {
					break;
				}
			}
		}
		if(length>sum/2) {
			printf("%d\n",sum);

		} else {
			printf("%d\n",length);
		}
	}
	return 0;
}