DFS POJ 1011 Sticks

来源:互联网 发布:vb教程 编辑:程序博客网 时间:2024/04/28 18:39

Sticks
Time Limit: 1000MS Memory Limit: 10000K
Total Submissions: 130321 Accepted: 30531
Description
George took sticks of the same length and cut them randomly until all parts became at most 50 units long. Now he wants to return sticks to the original state, but he forgot how many sticks he had originally and how long they were originally. Please help him and design a program which computes the smallest possible original length of those sticks. All lengths expressed in units are integers greater than zero.
Input
The input contains blocks of 2 lines. The first line contains the number of sticks parts after cutting, there are at most 64 sticks. The second line contains the lengths of those parts separated by the space. The last line of the file contains zero.
Output
The output should contains the smallest possible length of original sticks, one per line.
Sample Input
9
5 2 1 5 2 1 5 2 1
4
1 2 3 4
0
Sample Output
6
5
Source
Central Europe 1995

题意:给出n根小棒的长度stick[i],已知这n根小棒原本由若干根长度相同的长木棒(原棒)分解而来。求出原棒的最小可能长度。
深搜过程类似于DFS Network Saboteur poj 2531和DFS POJ 2362 Square ,都是将元素分为两个集合。
思路:dfs+剪枝。实现:求出总长度sum和小棒最长的长度max,则原棒可能的长度必在max~sum之间,然后从小到大枚举max~sum之间能被sum整除的长度len,用dfs求出所有的小棒能否拼凑成这个长度,如果可以,第一个len就是答案。
下面就是关键的了,就是这道题dfs的实现和剪枝的设计:
1.以一个小棒为开头,用dfs看看能否把这个小棒拼凑成len长,如果可以,用vis[i]记录下用过的小棒,然后继续以另外一个小棒为开头,以此类推。
2.小棒的长度从大到小排序,这个就不解释了。
3.如果当前最长的小棒不能拼成len长,那么就返回前一步,更改前一步的最长小棒的组合情况(这里不能是全部退出),不用再继续搜索下去了。
4.最重要的,就是比如说17,9,9,9,9,8,8,5,2……如果当前最长小棒为17,它与第一个9组合之后dfs发现不能拼成len,那么17就不用和后面所有的9组合了,而直接和8开始组合。
//132K 16MS

#include<iostream>#include<algorithm>#include<cstdio>#include<cstring>using namespace std;const int Max = 65;int n, aveLen, stick[Max];bool flag, used[Max];bool cmp(int a, int b){    return a > b;}void dfs(int sticksNum, int now_Len, int Now){//dfs功能:判断是否构成一根原棒;如果不能,就从Now开始枚举其后所有小棒的使用情况(使用和没使用),看能否构成一根原棒       // sticksNum为当前已被用过的小棒数,Now为当前要处理的小棒。    if(flag) return;    if(now_Len == 0){                    //当前长度为0,即从头开始拼接新棒        int k = 0;        while(used[k]) k ++;              //寻找当前最长小棒。        used[k] = true;        dfs(sticksNum + 1, stick[k], k + 1);        used[k] = false;        return;    }    if(now_Len == aveLen){                  //  当前长度为aveLen,即又拼凑成了一根原棒。        if(sticksNum == n) flag = true;        //  完成的标志:所有的n根小棒都有拼到了。        else dfs(sticksNum, 0, 0);//重新拼接下一根        return;    }    for(int i =Now; i < n; i ++)//从Now开始枚举其后所有小棒的使用情况(使用和没使用),看能否构成一根原棒         if(!used[i] && now_Len + stick[i] <= aveLen){            if(!used[i-1] && stick[i] == stick[i-1]) continue;//不重复搜索:最重要的剪枝。            used[i] = true;//使用当前小棒            dfs(sticksNum + 1, now_Len + stick[i], i + 1);//使用当前小棒后判断是否构成一根原棒;如果不能,就从 i + 1开始枚举其后所有小棒的使用情况(使用和没使用),看能否构成一根原棒            used[i] = false;//回溯        }}int main(){      freopen("input.txt","r",stdin);    freopen("output.txt","w",stdout);    while(scanf("%d", &n)!=EOF && n != 0){        int sum = 0;        flag = false;        for(int i = 0; i < n; i ++){            scanf("%d", &stick[i]);            sum += stick[i];        }        sort(stick, stick + n, cmp);     //  从大到小排序。        for(aveLen = stick[0]; aveLen < sum; aveLen ++)            if(sum % aveLen == 0){          //  枚举能被sum整除的长度。                memset(used, 0, sizeof(used));                dfs(0, 0, 0);                if(flag) break;            }        printf("%d\n", aveLen);    }    return 0;}
0 0