P1273 有线电视网

Posted 2020-12-25 lance1ot

P1273 有线电视网

就是一个树形分组背包模板。然后最后在、倒叙遍历一遍根节点的dp数组就可以了

#include<cstdio>
#include<algorithm>
#include<iostream>
#include<cstring>
using std::max;
const int maxn=3010;
struct node
{
    int point;
    int weight;
    int nxt;
};
node line[maxn<<1];
int head[maxn],tail;
void add(int a,int b,int c)
{
    line[++tail].point=b;
    line[tail].nxt=head[a];
    line[tail].weight=c;
    head[a]=tail;
}
int cost[maxn];
int son[maxn];
int f[maxn][maxn];
void dfs(int now)
{
    bool has_son=false;
    f[now][0]=0;
    for(int i=head[now];i;i=line[i].nxt)
    {
        has_son=true;
        dfs(line[i].point);
        son[now]+=son[line[i].point];
        for(int j=son[now];j>=0;j--)
            for(int k=0;j-k>=0&&k<=son[line[i].point];k++)
                f[now][j]=max(f[now][j],f[now][j-k]+f[line[i].point][k]-line[i].weight);
    }
    if(!has_son)
    {
        son[now]=1;
        f[now][1]=cost[now];
        return ;
    }
    return ;
}
int main()
{
    memset(f,-10,sizeof(f));
    int n,m;
    scanf("%d%d",&n,&m);
    for(int i=1;i<=n-m;i++)
    {
        int k,a,b;
        scanf("%d",&k);
        for(int j=1;j<=k;j++)
        {
            scanf("%d%d",&a,&b);
            add(i,a,b);
        }
    }
    for(int i=n-m+1;i<=n;i++)
        scanf("%d",&cost[i]);
    dfs(1);
    for(int i=son[1];i>=0;i--)
    {
        if(f[1][i]>=0)
        {
            printf("%d",i);
            return 0;
        }
    }
}