洛谷 P1273 有线电视网 题解

题面

按照常见树形背包定义状态：设dp[u][j]表示在以u为根的子树中，选择j个客户所能获得的最大收益。

状态转移：dp[u][j]=max(dp[u][j-k],dp[v][k]-w(u,v));

注意因为二维的是滚动数组，所以应该倒着循环。

#include <bits/stdc++.h>
using namespace std;
int n,m;
struct littlestar{
    int to;
    int nxt;
    int w;
}star[];
int head[],cnt;
void add(int u,int v,int w)
{
    star[++cnt].to=v;
    star[cnt].nxt=head[u];
    star[cnt].w=w;
    head[u]=cnt;
}
int money[];
int f[][];
int dfs(int u)
{
    if(u>n-m){
        f[u][]=money[u];
        f[u][]=;
        return ;
    }
    f[u][]=;
    int sum=;
    for(int i=head[u];i;i=star[i].nxt){
        int v=star[i].to;
        int tmp=dfs(v);
        sum+=tmp;
     for(int j=sum;j>=;j--){
            for(int k=;k<=min(j,tmp);k++){
                f[u][j]=max(f[u][j],f[u][j-k]+f[v][k]-star[i].w);
            }
        }
    }
    return sum;
}
int main()
{
    cin>>n>>m;
    for(int i=;i<=n-m;i++){
        int num;
        scanf("%d",&num);
        for(int j=;j<=num;j++){
            int v,w;
            scanf("%d%d",&v,&w);
            add(i,v,w);
        }
    }
    for(int i=n-m+;i<=n;i++){
        scanf("%d",&money[i]);
    }
    memset(f,0x9f,sizeof(f));
    for(int i=;i<=n;i++){
        f[][i]=;
    }
    dfs();
    for(int i=m;i>=;i--){
        if(f[][i]>=){
            cout<<i;
            return ;
        }
    }
}