木棒(回溯算法)

ac链接：木棒  来源：《算法竞赛进阶指南》 UVA307

题目：

乔治拿来一组等长的木棒，将它们随机地砍断，使得每一节木棍的长度都不超过 5050 个长度单位。

然后他又想把这些木棍恢复到为裁截前的状态，但忘记了初始时有多少木棒以及木棒的初始长度。

请你设计一个程序，帮助乔治计算木棒的可能最小长度。

每一节木棍的长度都用大于零的整数表示。

输入格式

输入包含多组数据，每组数据包括两行。

第一行是一个不超过 6464 的整数，表示砍断之后共有多少节木棍。

第二行是截断以后，所得到的各节木棍的长度。

在最后一组数据之后，是一个零。

输出格式

为每组数据，分别输出原始木棒的可能最小长度，每组数据占一行。

数据范围

数据保证每一节木棍的长度均不大于 5050。

输入样例：

9
5 2 1 5 2 1 5 2 1
4
1 2 3 4
0

输出样例：

6
5

思路分析：

①将木棍按从大到小进行排序

②枚举木棒长度，从给出木棍中最长的开始

③定义dfs，返回是否构成木棒，接收参数int c  已经复原的木棒数量，int cur 现长度 ，int start 开始下标

实现代码：

package oneQuestioinPerDay;

import java.util.Arrays;
import java.util.Scanner;

public class day15new {
	
	static int[] a;
	static boolean[] st;
	static int length,n,sum;
	
	static void swap(int i,int j) {
		int tmp = a[i];
		a[i] = a[j];
		a[j] = tmp;
	}
	
	static void reverse() {
		int i = 0;
		int j = n - 1;
		while (i <= j) {
			swap(i,j);
			i++;
			j--;
		}
	}
	
	static boolean dfs(int c,int cur,int start) {
		if (c * length == sum) return true;  //所有木棒找到了
		if (cur == length) return dfs(c+1,0,0); //长度够了，则开始找下个木棒
		if (cur > length) return false;  
		for (int i = start;i < n;i++) {
			if (st[i]) continue;  //剪枝，已经被用了，直接跳过
			st[i] = true;  
			if (dfs(c,cur+a[i],i+1)) return true; // 找木棍拼接
			st[i] = false; //回溯
			if (cur == 0) return false;
			if (cur+a[i] == length) return false; //最后一个木棍放入后可以达到length却
                                                    //没有成功则一定失败
			while(i < n-1&& a[i] == a[i+1]) {   // 跳过相同部分
				i++;
			}
		}
		return false;
	}
	
	public static void main(String[] args) {
		Scanner sc = new Scanner(System.in);
		n = sc.nextInt();
		while(n != 0) {
			a = new int[n];
			st = new boolean[n];
			length = sum = 0;
			
			for (int i = 0;i < n;i++) {
				a[i] = sc.nextInt();
				length = Math.max(length, a[i]);
				sum += a[i];
			}
			Arrays.sort(a);
			reverse();
			while (length <= sum) {
				if (sum % length == 0 && dfs(0,0,0)) break;
				length++;
			}
			System.out.println(length);
			n = sc.nextInt();
		}
	}
}