依赖背包优化——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