poj 1155 TELE （树形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的费用。最后一行依次表示所有用户为观看比赛而准备支付的钱数。

输出格式：

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

思路：树形DP DP[i][j]表示：以第i个节点的子树，总共使j个客户端接收到信息的最大报酬(可以为负数)。状态转移方程：dp[u][j+k]=max(dp[u][j+k],dp[u][j]+dp[v][k]-edge[i].w)。

#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
#include <cmath>
using namespace std;

struct node{
int to, w, next;
}ed[10005];

int n ,m, idc;
int num[3005],head[3005],dp[3005][3005],tem[3005];

void adde(int u, int v, int w){
ed[++idc].to = v;
ed[idc].w = w;
ed[idc].next = head[u];
head[u] = idc;
}

void dfs(int root){
for(int k=head[root]; k; k=ed[k].next){
int v = ed[k].to;
dfs(v);
for(int j=0; j<=num[root]; j++)
tem[j] = dp[root][j];//没有连v时的代价
for(int j=0; j<=num[root]; j++){
for(int p=1; p<=num[v]; p++)
dp[root][p+j] = max(dp[root][p+j],tem[j]+dp[v][p]-ed[k].w);//连v
}
num[root] += num[v];//最多可连客户数
}
}

int main(){
int k,a,b;
scanf("%d%d", &n, &m);
memset(head,0,sizeof(head));
for(int i=1; i<=n-m; i++){
scanf("%d", &k);
num[i] = 0;//边界，最多连客户数
for(int j=1; j<=k; j++){
scanf("%d%d", &a, &b);
adde(i, a, b);
}
}
for(int i=1; i<=n; i++)
for(int j=1; j<=m; j++)
dp[i][j] = -10000000;//利润可能为负，所以初始值要小一点，不到客户代价为0
for(int i=n-m+1; i<=n; i++){
num[i] = 1;//可连接一个客户
scanf("%d", &dp[i][1]);//利润
}
dfs(1);
for(int i=m; i>=0; i--){
if(dp[1][i] >= 0){//不亏本
printf("%d\n",i);
break;
}
}
return 0;
}