树形背包模版-洛谷P1273 有线电视网

https://www.luogu.org/problem/show?pid=1273
其实就是一个背包嘛,每个终端都是一个物品,体积全是1;
网上的题解很多,我copy一段
http://blog.csdn.net/qwsin/article/details/50954669
dp[i][j]表示i节点在其后代中选了j个的最大收入
dp[i][j]=max(dp[i][j],dp[i][j-k]+dp[v][k]-w)
相信大家都看得懂这个公示,但是也许你会有所疑惑;
这个更新方程需要调用它自己!
那我怎么可以保证dp[i][j-k]中我没有包含v这个子节点的情况;
还有为什么j要倒着枚举????;
其实呢,这个只不过是01背包的树版;
这相当与一维的01背包;
其实原先应该是dp[i][j][k]
表示第i个节点的前j个字节点中,我在第j个字节点选k个的最大最优情况;
把j这一个压缩掉就变成了一开始我们说的那个dp方程了;
所以呢,显然的一开始方程的j要倒着枚举,原理和一维的01背包是一样的;

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cstdlib>
#include<cmath>
#define Ll long long
using namespace std;
struct cs{
    int to,next,vv;
}a[3001];
int head[3001],v[3001],f[3001][3001];
int n,m,ll,x,y,z,nn;
void init(int x,int y,int z){
    ll++;
    a[ll].to=y;
    a[ll].vv=z;
    a[ll].next=head[x];
    head[x]=ll;
}
int dfs(int x){//返回x节点包含了几个叶子节点 
    if(!head[x]){
        f[x][1]=v[x];return 1;
    }
    int sum=0,son=0;
    for(int k=head[x];k;k=a[k].next){
        son=dfs(a[k].to);
        sum+=son;
        for(int j=sum;j;j--)
        for(int i=1;i<=min(son,j);i++)
            f[x][j]=max(f[x][j],f[x][j-i]+f[a[k].to][i]-a[k].vv);
    }
    return sum;
}
int main()
{
    memset(f,-63,sizeof f);
    scanf("%d%d",&n,&m);
    for(int i=1;i<=n-m;i++){
        scanf("%d",&z);nn++;
        for(int i=1;i<=z;i++){
            scanf("%d%d",&x,&y);
            init(nn,x,y);
        }
    }
    for(int i=1;i<=m;i++)scanf("%d",&v[++nn]);
    for(int i=0;i<=n;i++)f[i][0]=0;
    nn=dfs(1);
    for(int i=m;i>=0;i--)
        if(f[1][i]>=0){
            printf("%d",i);return 0;
        }
}
posted @ 2017-02-19 18:17  largecube233  阅读(101)  评论(0编辑  收藏  举报