菜鸟学算法之 POJ1011 Sticks

　　这道题真是太经典了，学 dfs 必须理解透彻．

/*
* 参考了网上大神的思路，代码均为原创．
* 这道题在　dfs　类型中的经典地位无可动摇，包含很多的剪枝，每一个剪枝都需要花心思好好想．
* AC : 132K 16ms
*/
#include<iostream>
#include<stdio.h>
#include<string.h>
#include<algorithm>
using namespace std;
int a[66];          //　存储所有小棍子的长度
int n;              // 小棍子的个数
bool flag;          // flag=true:成功还原所有棍子
int ori_len;        // 原本的棍子长度
int ori_num;        // 原本的棍子数目
bool visited[66];   // visited[i]=true表示第i根棍子已经被使用了，不能重复使用

inline bool compare(int a , int b)  // 用于sort()降序排列
{
    return a > b;
}

// len:已经拼接起来的长度　num:已经拼接了多少根棍子 pos:本次搜索的起始位置
// 如果a[i]被选中，那么接下来从 a[i+1]开始搜索，达到剪枝的目的(第1处剪枝）
void dfs( int len , int num , int pos )
{
    // dfs的第一步，判断结束条件,是否已经成功还原所有棍子
    if( true == flag )
        return;
    // 是否还原了一根棍子
    if( len == ori_len )            //　成功地还原了一根棍子
    {
        if( num + 1 == ori_num )    // 成功地还原了所有的棍子
        {
            flag = true;
            return;
        }
        dfs( 0 , num + 1 , 0 );   // 新开一根棍子，继续搜索
        return;
    }
    // 搜索所有棍子，但是对于多根相同长度的棍子，如果第一根棍子被使用了之后，则不再使用相同长度的棍子（第２处剪枝）
    // 这个剪枝不太好理解，我也是想了好半天才想清楚．
    int prev_length = - 1;  // 记录当前被使用的棍子的长度
    for(int i=pos;i<n;i++)
    {
        if( true == visited[i] || a[i] == prev_length ) // 已经被使用过
            continue;
        if( len + a[i] <= ori_len ) // 可以使用第 i　根棍子的条件
        {
            prev_length = a[i]; //　正式地使用第 i　根棍子
            visited[i] = true;  // 标记
            dfs( len + a[i] , num , i +1 ); // 继续搜索

            if( false == flag )     // 如果使用了第 i　根棍子，但是获得了失败
            　　visited[i] = false; // 则不使用第 i 根棍子
        }

        // 第　３　处剪枝，如果新开的＂棒子＂的第一根合适的＂棍子＂无法与其他棍子组合，说明该＂棒子＂不合适
        // 这个剪枝来自网上的大神，我想了好半天，终于有自己的理解：
        // 假定：所有的小棍子能够拼接成 ori_num 根　ori_len　长度的＂棒子＂，那么，每一根棍子都必然能够与若干根其他的棍子一起，
        // 拼接成为ori_len长度的＂棒子＂，但是现在，第一根合适的棍子无法找到剩余的棍子来拼接，只能说明，该＂棒子＂不合适，舍弃！
        if( 0 == len && false == flag )
            break;
    }
}

int main()
{
    int i,j,k;
    while( scanf("%d",&n) && n != 0 )
    {
       int sum_len = 0; // 所有棍子的总长度
       flag = false;    // 初始化 flag
       memset( visited , 0 , sizeof(visited) ); // 初始化visited

       for(i=0;i<n;i++)
       {
          scanf("%d",&a[i]);
          sum_len += a[i];
       }
       // 排序：第　４　处剪枝，按照长度从大到小排序，可以尽量地缩减搜索空间，从而加快速度
       sort( a , a + n , compare );
       int max_len = a[0];  // 获取最大的长度，作为搜索的最小值
       // ＤＦＳ搜索
       for(ori_len =max_len;ori_len <= sum_len/2 ;ori_len++)
       {
            if( sum_len % ori_len != 0 )    // ori_len 必定被 sum_len 整除（剪枝５）
                continue;
            ori_num = sum_len / ori_len;
            dfs( 0 , 0 , 0 );   // 开始DFS
            if( true == flag )
                break;
       }
       if( true == flag )
          printf("%d\n",ori_len);
       else
          printf("%d\n",sum_len);
    }
    return 0;
}