Vijos1159 岳麓山上打水(迭代加深 + DP)

题意分析

其实一开始写了一个假算法,看出来是完全背包了,就完全背包+记录路径,一直WA,后来看了题解才明白是迭代加深和DP。

代码总览

#include
using namespace std;
const int nmax = 101;
int a[nmax],ans[nmax];
int dp[20001];
int p,q,sum;
bool isok= false;
bool getdp(int c){
    memset(dp,0,sizeof dp);
    dp[sum] = 1;
    for(int j = 1;j<=c;++j){
        for(int i = sum;i<=q;++i)
            if(dp[i - ans[j]]) dp[i] = 1;
    }
    if(dp[q]){
        printf("%d ",c);
        for(int i = 1;i<=c;++i)
            printf("%d ",ans[i]);
        return true;
    }
    return false;
}
void dfs(int dep,int rt, int c){ // 从rt开始 选择c种 然后看是否能够满足题目要求
    if(isok) return;
    ans[dep] = a[rt];
    sum += a[rt];
    if(dep == c){
        if(getdp(c)){
            isok = true;
            return;
        }
    }else
        for(int i = rt+1;i<=p;++i)
            dfs(dep+1,i,c);
    sum-=a[rt];
}
int main(){
    scanf("%d %d",&q,&p);
    for(int i = 1;i<=p;++i) scanf("%d",&a[i]);
    sort(a+1,a+1+p);
    for(int i = 1;i<=p;++i){
        for(int j = 1;j// 至少选择1种
            dfs(1,j,i);
            if(isok) return 0;
        }
    }
    return 0;
}

你可能感兴趣的:(算法---动态规划)