洛谷 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;}