洛谷 P1120 小木棍 ［数据加强版］

题目来源：https://www.luogu.org/problem/show?pid=1120

在各网站提交了不下20次，终于AC了.......

一开始是用贪心做的，并没有考虑已拼接成的木棍在回溯时还会再拆开重组。程序跑得飞快但总WA几个点。在各网站提交都是WA。

后来看了题解，做法都和我不一样，但我却找不出错误。

生成了几组数据用题解对拍，才找出错误......

但会正解后，又TLE了几次，最后在题解的指引下剪了几个枝才AC。

先排序，再枚举长度，木棍从大到小依次验证是否可拼接成所枚举的长度，一旦不可行则直接回溯。

几个剪枝：

1.如果某个长度不可行，则与之相同长度的必然不行，因而不用去搜。

2.一旦拼接成功，直接输出结果，退出程序。

3.枚举木棍长度时一定要保证木棍总长能够被其整除，否则不必验证。

不同网站对>50的长度要求不同，洛谷上要求滤掉>50的长度。

代码：

#include <iostream>
#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <algorithm>
#include <queue>
using namespace std;
int n;
int a[100];
int cnt=0;
bool vis[100];
bool dfs(int num,int len,int s,int pos)
{
    int ok=1;
    if(num==1){printf("%d",s);exit(0);}
    int p=-1;
    for(int i=pos;i>=1;i--)
    {
        if(!vis[i]&&a[i]!=p)
        {
            vis[i]=1;
            if(len+a[i]<s)
            {
                if(dfs(num,len+a[i],s,i-1)){printf("%d",s);exit(0);}
                if(len==0){vis[i]=0;return 0;}               /
            }
            if(len+a[i]==s)
            {
                if(dfs(num-1,0,s,cnt)){printf("%d",s);exit(0);}
                else{vis[i]=0;break;}
            }
            vis[i]=0;
            p=a[i];
        }
    }
    return 0;
}
int main()
{
    scanf("%d",&n);
    int l=0;
    int sum=0;
    for(int i=1;i<=n;i++)
    {
        int c;scanf("%d",&c);
        if(c<=50)
        {
            sum+=c;
            a[++cnt]=c;
            l=max(l,c);
        }
    }
    sort(a+1,a+1+cnt);
    for(int i=l;;i++)
    {
        if(sum%i==0)
        {
            memset(vis,0,sizeof(vis));
            if(dfs(sum/i,0,i,cnt))
            {
                printf("%d",i);
                return 0;
            }
        }
    }
    return 0;
}