B23 DFS剪枝 小木棍

视频链接：124 小木棍 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++){
    if(sum%len) continue; //不合法就跳过
    cnt=sum/len;
    dfs(1,0,1); //第一次拼好即答案
  }
}

 

#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
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;
    used[i]=0; //拼接a[i]失败，恢复现场，再尝试a[i+1]
    
    if(cur==0)return false;       //cut1:拼接第1个，失败回溯
    if(cur+a[i]==len)return false;//cut2:拼接最后1个，失败回溯
    while(i<n&&a[i]==a[i+1]) i++; //cut3:跳过与a[i]等长的元素
  }
  return false; //尝试完后面的所有的数，失败回溯
}
int main(){
  scanf("%d",&n); int sum=0;
  for(int i=1; i<=n; ++i)scanf("%d",&a[i]),sum+=a[i];
  sort(a+1,a+1+n); reverse(a+1,a+1+n); //优化搜索顺序
  for(len=a[1]; ;++len){
    if(sum%len) continue; //不合法就跳过
    cnt=sum/len;
    if(dfs(1,0,1)) break; //第一次拼好即答案
  }
  printf("%d",len);
}