【洛谷】有线电视网-树形DP

传送门：洛谷-有线电视网

---

题意

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

---

输入

输入文件的第一行包含两个用空格隔开的整数N和M，其中2≤N≤3000，1≤M≤N-1，N为整个有线电视网的结点总数，M为用户终端的数量。
第一个转播站即树的根结点编号为1，其他的转播站编号为2到N-M，用户终端编号为N-M+1到N。
接下来的N-M行每行表示—个转播站的数据，第i+1行表示第i个转播站的数据，其格式如下：
K A1 C1 A2 C2 … Ak Ck
K表示该转播站下接K个结点(转播站或用户)，每个结点对应一对整数A与C，A表示结点编号，C表示从当前转播站传输信号到结点A的费用。最后一行依次表示所有用户为观看比赛而准备支付的钱数。

---

输出

输出文件仅一行，包含一个整数，表示上述问题所要求的最大用户数。

---

题解

蒟蒻看了题解才懂的。表示代码能力太弱了orzzzz
设F[i][j]表示以i为根结点的子树总共让j个用户看到比赛的钱数的最大值。满足F[1][j]的非负的最大j就是答案（注意这里找最大非负要从m倒着找，不然很可能再某个次大非负值与最大非负值之间有很多负值，无法判断答案）。
更新f[i][j]我们直接在每个子结点dfs后循环一遍加（本蒟蒻就是这里没想到这么更新啊，没想到这么直接）。

---

代码

#include<cstdio>
#include<iostream>
#include<cstring>
#include<algorithm>
#include<vector>
#define INF 0x7fffffff 
using namespace std;
const int N=3e3+10;
const int M=5e4+10;

int n,m,tot,op;
int head[N],to[M],nxt[M],w[M];
int val[N],dp[N][N],mx[N],put[N];

vector<int>ch;

inline void lk(int u,int v,int val)
{
    to[++tot]=v;nxt[tot]=head[u];head[u]=tot;w[tot]=val;
}

inline int max(int x,int y)
{
    return x>y? x:y;
}

inline int dfs(int x)
{
    if(x>(n-m)){
        dp[x][1]=val[x];
        return 1;
    }
    int sum=0;
    for(int i=head[x];i;i=nxt[i]){
        int e=to[i];
        int now=dfs(e);
        for(int kk=0;kk<=sum;kk++) put[kk]=dp[x][kk];
        for(int kk=0;kk<=sum;kk++)
          for(int j=1;j<=now;j++)
           dp[x][kk+j]=max(dp[x][kk+j],put[kk]+dp[e][j]-w[i]);
        sum+=now;
    }
    return sum;
}

int main(){
    scanf("%d%d",&n,&m);
    for(int i=1;i<=n;i++)
       for(int j=1;j<=m;j++)
        dp[i][j]=-INF;
    for(int i=1;i<=(n-m);i++){
        int c,a;
        scanf("%d",&op);
        while(op--){
            scanf("%d%d",&a,&c);
            lk(i,a,c);
        }
    }
    for(int i=(n-m)+1;i<=n;i++) scanf("%d",&val[i]);
    dfs(1);int ans=m;
    for(;dp[1][ans]<0;ans--);
    printf("%d\n",ans);
    return 0;
}