B23 DFS剪枝 小木棍

视频链接：B23 DFS剪枝 小木棍_哔哩哔哩_bilibili

 

Luogu P1120 小木棍

 

 

 

 

#include <iostream>
#include <cstring>
#include <algorithm>
using namespace std;

const int N=70;
int n,len,cnt;
int a[N],used[N];

//当前拼接第u根原始木棍，已拼接的长度为cur，
//下一段开始拼接的小木棍为start
void dfs(int u, int cur, int start){
  if(u>cnt){printf("%d",len);exit(0);} //拼好全部，立即结束
  if(cur==len){dfs(u+1,0,1);return;} //拼好第u根，再拼下一根
  for(int i=start; i<=n; i++){
    if(used[i]||cur+a[i]>len) continue; //不合法就跳过
    used[i]=1;
    dfs(u,cur+a[i],i+1);
    used[i]=0; //拼接a[i]失败，恢复现场，再尝试a[i+1]
    
    if(cur==0) return;          //cut1:拼接第1个，失败回溯
    if(cur+a[i]==len) return;   //cut2:拼接最后1个，失败回溯
    while(i<n&&a[i]==a[i+1])i++;//cut3:跳过与a[i]等长的元素
  }
}
int main(){
  cin>>n; int sum=0;
  for(int i=1; i<=n; i++)cin>>a[i],sum+=a[i];
  sort(a+1,a+1+n); reverse(a+1,a+1+n); //优化搜索顺序
  for(len=a[1];len<=sum;len++){
    if(sum%len) continue; //不合法就跳过
    cnt=sum/len;
    dfs(1,0,1); //第一次拼好即答案
  }
}

 

#include<bits/stdc++.h>
using namespace std;

const int N=70;
int n,a[N];
int len,cnt,used[N];

//尝试拼接长度为len，数量为cnt的原始木棍
//现在拼接第u根原始木棍，已拼接的长度为cur，下一段开始的小木棍为start
bool dfs(int u,int cur,int start){
  if(u>cnt) return true;            //全部拼好，搜索成功
  if(cur==len) return dfs(u+1,0,1); //已拼好第u根，再拼下一根
  for(int i=start; i<=n; i++){
    if(used[i]||cur+a[i]>len) continue;
    used[i]=1;
    if(dfs(u,cur+a[i],i+1)) return true; //如果拼接a[i]成功返回true
    used[i]=0;
    if(cur==0) return false; //第1个失败
    if(cur+a[i]==len) return false; //最后1个失败
    while(i<=n&&a[i]==a[i+1]) i++; //a[i]失败，则与之等长的必失败
  }
  return false;
}
int main(){
  cin>>n;
  int sum=0;
  for(int i=1;i<=n;i++) cin>>a[i],sum+=a[i];
  sort(a+1,a+1+n);  
  reverse(a+1,a+1+n); //优化搜索顺序
  for(len=a[1];len<=sum;len++){
    if(sum%len) continue;
    cnt=sum/len;
    if(dfs(1,0,1)) break; //最小长度即答案
  }
  cout<<len<<endl;
}