T1442 小木棍（信息学奥赛一本通）

【问题】

乔治有一些同样长的小木棍，他把这些木棍随意砍成几段，直到每段的长都不超过50。现在，他想把小木棍拼接成原来的样子，但是却忘记了自己开始时有多少根木棍和它们的长度。给出每段小木棍的长度，编程帮他找出原始木棍的最小可能长度。

【输入】

第一行为一个单独的整数N表示砍过以后的小木棍的总数，其中N≤60，第二行为N个用空个隔开的正整数，表示N根小木棍的长度。

【输出】

仅一行，表示要求的原始木棍的最小可能长度。

【源代码】

不知道为啥这个代码在信息学奥赛一本通网站上过不了

#include<bits/stdc++.h>
using namespace std;
const int N = 70;
int n,sum,len,tot;
int a[N],vis[N];
bool cmp(int u,int v)
{
    return u > v ;
}
// 第k根，还需要now的长度，枚举第pos根木棍
bool dfs(int k,int now,int pos)
{
    if(k == tot + 1) return true; //已经拼完了前tot根，所以返回答案
    if(now == 0)
    {
        // 如果当前木棍已经拼接好，接着拼下一根，直到拼完tot根为止
        return dfs(k+1,len,1);
    }
    // 剪枝2，如果当前用了第i个，则从i+1根开始拼
    for(int i=pos; i<=n; i++)
    {
        if(!vis[i] && a[i]<=now)      // 选择当前木棍，必须是没选中过，同时适合拼接下去
        {
            vis[i] = 1 ;
            if(dfs(k,now-a[i],i+1)) return true;
            vis[i] = 0 ;
            if(now == len || now == a[i]) //如果拼不出来直接返回，原因是不可能再继续下去
                return false;
            while(a[i] == a[i+1]) i++;    //相同长度不要多次搜索
        }
    }
    return false;
}
int main()
{
    cin >> n;
    for(int i=1; i<=n; i++)
    {
        cin >> a[i];
        sum += a[i] ;
    }
    sort(a+1,a+1+n,cmp );
    for(int i=a[1]; i<=sum; i++)
    {
        if(sum%i==0)
        {
            memset(vis,0,sizeof vis) ;
            len = i ;           // 设为全局变量，木棍的长度
            tot = sum/i;        // 木棍的数目
            if(dfs(1,len,1))
            {
                cout << len <<endl;
                return 0;
            }
        }
    }
    return 0;
}

AC代码

#include<bits/stdc++.h>
#define INF 0x3f3f3f3f
#define LL long long
const int N = 1000+5;
using namespace std;
int a[N];
bool vis[N];
int cnt;
bool cmp(int x,int y){ return x>y; }
bool dfs(int k,int step,int rest,int len){
//k为已经拼接好的个数,step为上一个的编号,rest为剩余的长度,len为木棍的长度
    if(k==cnt+1 && rest==0)//拼接好个数为要求数且无剩余
        return true;
    else if(k==cnt+1)//拼接好个数为要求数但有剩余
        return false;
    else if(rest==0){//拼接好个数不为要求数但无剩余，重新开始
        rest=len;//剩余数变为长度
        step=0;//编号归零
    }
    for(int i=step+1;i<=cnt;i++){
        if(!vis[i]){
            if(rest-a[i]>=0){//保证剩余值不为负数
                vis[i]=true;
                if(dfs(k+1,i,rest-a[i],len))
                    return true;
                vis[i]=false;
                if(a[i]==rest || len==rest)//头尾剪枝，此时已在回溯之后，需要判断头尾两种情况
                    break;
                while(a[i]==a[i+1])//去重剪枝，用当前长度搜索无结果时，对同样长度的可以忽略
                    i++;
            }
        }
    }
    return 0;
}
int main(){
    int n;
    cin >> n;
    int sum=0;
    for(int i=1;i<=n;i++){
        int x;
        cin >> x;
        if(x<=50){
            a[++cnt]=x;
            sum+=x;
        }
    }
    sort(a+1,a+1+cnt,cmp);
    for(int i=a[1];i<=sum;i++){
        if(sum%i==0){
            if(dfs(1,0,i,i)){
                cout << i <<endl;
                break;
            }
        }
    }
    return 0;
}