洛谷P1120 小木棍 ［数据加强版］【搜索+剪枝】

题目描述 https://www.luogu.org/problemnew/show/P1120
将木棍排序，从大到小枚举，再加各种剪枝。

#include<iostream>
#include<cstring>
#include<cstdio>
#include<algorithm>
using namespace std;
int a[70],fl[70],next[70],n,m,minn,sum,f,len,cnt;
bool cmp(const int &a,const int &b)
{
	return a>b;
}
void dfs(int now,int last,int rest)//now表示第now跟木棍
{ //last是第now跟木棍的上一个编号 rest表示第now跟还需要的长度 
    int i;
	if(!rest)//这根拼好 可以拼下一根 
	{ 
        if(now==m){f=1; return;}//可行 返回输出 
        for(i=1;i<=cnt;i++) if(!fl[i]) break;
        fl[i]=1; 
        dfs(now+1,i,len-a[i]);
        fl[i]=0;
        if(f) return; 
    }
    for(int i=last+1;i<=cnt;i++)
	  if(!fl[i]&&a[i]<=rest)
	  {
         fl[i]=1;
         dfs(now,i,rest-a[i]);
         fl[i]=0;
         if(f) return;
		 if(rest==a[i]||rest==len) return;//剪枝
         i=next[i];//这种情况不可行 后面与它相等的木棍都不可行 
         if(i==n) return;
      }
}
int main()
{  
   scanf("%d",&n);
   for(int i=1;i<=n;i++)
   {
   	  int z;
   	  scanf("%d",&z);
   	  if(z>50) continue;
   	  a[++cnt]=z;
   	  minn=max(minn,z);
   	  sum+=z;
   }
   sort(a+1,a+cnt+1,cmp);
   next[cnt]=cnt;
   for(int i=cnt-1;i>0;i--)
   {
      if(a[i]==a[i+1]) next[i]=next[i+1];
      else next[i]=i;
   }//预处理出相等的木棍
   for(int i=minn;i<=sum;i++)
   {
   	  if(sum%i!=0) continue;
   	  memset(fl,0,sizeof(fl));
   	  len=i; m=sum/i; f=0;
   	  fl[1]=1;
   	  dfs(1,1,len-a[1]);
   	  if(f==1){ printf("%d",len); break;}
   }
 	
  return 0;	
}