Luogu P1273 有线电视网 题解

Describe

题目链接

某收费有线电视网计划转播一场重要的足球比赛。他们的转播网和用户终端构成一棵树状结构，这棵树的根结点位于足球比赛的现场，树叶为各个用户终端，其他中转站为该树的内部节点。 从转播站到转播站以及从转播站到所有用户终端的信号传输费用都是已知的，一场转播的总费用等于传输信号的费用总和。 现在每个用户都准备了一笔费用想观看这场精彩的足球比赛，有线电视网有权决定给哪些用户提供信号而不给哪些用户提供信号。 写一个程序找出一个方案使得有线电视网在不亏本的情况下使观看转播的用户尽可能多。

Solution

类似于分组背包。 设$f[i][j]$表示第$i$个子树，使用$j$个的最大钱数。 $f[i][j]=$$max(f[i][j-k]$$+f[to][k]-$$w[edge])$ 那么转移就好了

Code

#include<bits/stdc++.h>
using namespace std;
int n,m,fir[3010],nxt[3010],w[3010],son[3010],tot,v[3010],f[3010][3010];
inline void add(int x,int y,int z){++tot;nxt[tot]=fir[x];fir[x]=tot;son[tot]=y;w[tot]=z;}
inline int DFS(int x){
    if(n-m+1<=x&&x<=n){
        f[x][1]=v[x];
        return 1;
    }
    int sum=0;
    for(int tmp,to,i=fir[x];i;i=nxt[i]){
        to=son[i];
        tmp=DFS(to);
        sum+=tmp;
        for(int j=sum;j;j--){
            for(int k=1;k<=tmp;k++){
                if(j-k>=0) f[x][j]=max(f[x][j],f[x][j-k]+f[to][k]-w[i]);
            }
        }
    }
    return sum;
}
int main(){
    scanf("%d%d",&n,&m);
    for(int x,y,z,i=1;i<=n-m;i++){
        scanf("%d",&x);
        for(;x;x--) scanf("%d%d",&y,&z),add(i,y,z);
    }
    for(int i=n-m+1;i<=n;i++) scanf("%d",&v[i]);
    memset(f,-63,sizeof(f));
    for(int i=1;i<=n;i++) f[i][0]=0;
    DFS(1);
    for(int i=m;i>=1;i--){
        if(f[1][i]>=0){printf("%d ",i);break;}
    }
    return 0;
}