【USACO5-3-2】量取牛奶Milk Measuring dp+dfs

原题

#include<iostream>
#include<algorithm>
#include<cstdio>
#include<cstring>
#include<cctype>
using namespace std;
int ans[1010],res[1010],c[1010];
int n,m;
bool flag;
void dfs(int max_d,int d,int r,int cnt){
    if (max_d==d){//如果到了最多的个数 
        if (!r){
            flag=true;
            memcpy(ans,res,sizeof(res));
        }
        return;
    }
    if(cnt>=m||c[cnt]>ans[d]) return;	
    res[d]=c[cnt];
    for (int i=1;i*c[cnt]<=r;i++) dfs(max_d,d+1,r-i*c[cnt],cnt+1);
    dfs(max_d,d,r,cnt+1);
}
int main(){
    scanf("%d%d",&n,&m);
    for (int i=0;i<m;i++) scanf("%d",&c[i]),ans[i]=0x3f3f3f3f;
    sort(c,c+m);
    for (int i=1;i<=m;i++){
        flag=false;
        dfs(i,0,n,0);//i表示这种情况下集合元素最多为i 
        if (flag){
            printf("%d",i);
            for (int j=0;j<i;j++) printf(" %d",ans[j]);
            break;
        }
    }
    return 0;
}