依赖背包优化——ural1018，金明的预算方案

经典题了，网上博客一大堆O(nCC)的做法，其实是可以将复杂度降到O(nC)的

参考依赖背包优化（泛化物品的并）

根据背包九讲，求两个泛化物品的和复杂度是O(CC)的，所以依赖背包暴力求解的复杂度是O(nCC)

然后对其进行优化，考虑 F[u][j] 用来表示以结点u为根的子树(不包括u)体积为j的全局最大价值
枚举u的儿子v，求F[v][j]数组，因为要求以v为根的子树，必须先选v，所以把v强行塞到背包里
即通过F[v][j]=F[u][j]+W[v] (j<=C-V[v])这个赋值来表示j子树的最大体积不会超过C-V[v]，并且已经选了v
然后dfs求出F[v][j]的值，此时按照定义已经得到了一个新的泛化物品，即一定包含v，并且可能包含v子树元素与前面几棵子树元素的背包，
我们拿这个背包和之前的背包(即一定不包含v的子树)进行合并，去更新F[u],F[u][j]=max(F[u][j],F[v][j-V[v]])
即F[u]在体积为j的时候要么保持原来的值，不取v子树的任何物品，要么取一些v子树的物品，那么显然v物品也要被取

即每次处理完一个u的子树v后，我们就得到了一个和当前泛化物品有交集(或者说是严格的后者包含前者，因为加了一整棵v的子树(除了v没加)进去)的新的泛化背包，我们对其求并，复杂度是O(C)，因为对于每个体积，我们只要求两者在该体积下的最大价值即可，当然新的泛化物品因为要强行塞入v，所以最后对应的体积也会偏移V[v]、

加深理解：一整棵子树v是如何逐步加入到决策集合中的。在dfs进入v子树后，会再遍历v的子树w，然后把w强行塞入背包，塞入背包说明前面一定会有一些物品x被挤出必选v的最佳决策集合中，那么回溯的时候我们考虑是选w还是选择x更优，最后回溯到v的时候，整个v的子树已经决策好了。

#include<bits/stdc++.h>
using namespace std;
#define N 205
struct Edge{int to,nxt,w;}e[N<<1];
int head[N],tot,n,q;
void init(){memset(head,-1,sizeof head);tot=0;}
void add(int u,int v,int w){
    e[tot].w=w;e[tot].to=v;e[tot].nxt=head[u];head[u]=tot++;
}
int F[N][N];
void dfs(int u,int pre,int C){
    for(int i=head[u];i!=-1;i=e[i].nxt){
        int v=e[i].to;
        if(v==pre)continue;
        for(int j=0;j<=C-1;j++)
            F[v][j]=F[u][j]+e[i].w;
        dfs(v,u,C-1);
        for(int j=1;j<=C;j++)
            F[u][j]=max(F[u][j],F[v][j-1]);
    } 
}

int main(){
    while(cin>>n>>q){
        init();
        for(int i=1;i<n;i++){
            int u,v,w;
            scanf("%d%d%d",&u,&v,&w);
            add(v,u,w);add(u,v,w);
        }
        memset(F,0,sizeof F);
        dfs(1,0,q);
        cout<<F[1][q]<<endl;
    }
}

 这个题要加一个虚根，用暴力显然是过不了的。所以要优化后的依赖背包来做

#include<bits/stdc++.h>
using namespace std;
#define N 65
#define M 32005
#define ll long long
struct Edge{int to,nxt;}e[N<<1];
int head[N],tot;
void init(){memset(head,-1,sizeof head);tot=0;}
void add(int u,int v){
    e[tot].to=v;e[tot].nxt=head[u];head[u]=tot++;
}

int dp[N][M],V[N],W[N],n,C;
void dfs(int u,int pre,int C){
    for(int i=head[u];i!=-1;i=e[i].nxt){
        int v=e[i].to;
        if(v==pre)continue;
        for(int j=0;j<=C-V[v];j++)
            dp[v][j]=dp[u][j]+V[v]*W[v];
        dfs(v,u,C-V[v]);
        for(int j=V[v];j<=C;j++)
            dp[u][j]=max(dp[u][j],dp[v][j-V[v]]);
    }
}

int main(){
    while(cin>>C>>n){
    init();
    for(int i=1;i<=n;i++){
        int fa;
        scanf("%d%d%d",&V[i],&W[i],&fa);
        add(i,fa);add(fa,i);//增加一个0结点 
    }
    memset(dp,0,sizeof dp);
    dfs(0,0,C);
    int ans=0;
    for(int j=0;j<=C;j++)ans=max(ans,dp[0][j]);
    cout<<ans<<endl;
    }
}

 

转载于:https://www.cnblogs.com/zsben991126/p/11375659.html