AcWing 167. 木棒

分析

代码

#include <cstring>
#include <algorithm>
#include <iostream>
using namespace std;
const int N=70;
bool st[N];
int n;
int w[N];
int len;
int sum;
bool dfs(int u,int cnt,int start){
    if(u*len==sum)return true;//如果所有的木棒全部组成完了，就return true
    if(cnt==len)return dfs(u+1,0,0);//如果当前的木棒全部组成完了，就return true
    for(int i=start;i<n;i++){
        if(st[i])continue;
        if(w[i]+cnt>len)continue;
        st[i]=true;
        if(dfs(u,cnt+w[i],i+1))return true;
        st[i]=false;
        if(!cnt||cnt+w[i]==len)return false;//第一个木棍或者最后一个木棍组成失败了，就失败了
        int j=i;
        while(j<n&&w[j]==w[i])j++;//如果第i个木棍失败了，那么和该木棍等长的木棍也失败，所以直接略去
        i=j-1;
    }
    return false;
}
int main(){
    while(cin >> n,n){
        memset(st,false,sizeof st);
        sum=0;
        len=0;
        for(int i=0;i<n;i++){
            cin >> w[i];
            sum+=w[i];
        }
        sort(w, w + n);//从小到大排序，减小分支
        reverse(w, w + n);
        len=w[0];//len从所有木棍里最大长度开始枚举
        while(len){
            if(sum % len == 0&&dfs(0,0,0)){//如果枚举长度是总长度的因子时才有可能成功
                cout << len <<"\n";
                break;
            }
            len++;
        }
    }
}