POJ 1155 TELE （树形背包）

题目链接：http://poj.org/problem?id=1155

题目大意：给出一棵n个结点的树，每个叶子结点有一个权值，每条边也有一个权值。求一个叶子节点数最大的集合，使得该集合中所有点的权值和减去这些点到达根结点所经过的边（重复了只算一条）的权值之差为非负值。

思路：另dp(i, j)表示从以i为根结点的子树中选j个叶子结点所剩金钱的最大值，则dp(u, j) = max{ dp(u, k) + dp(v, j-k) - cost(u, v)}，其中dp(u, k)不包含结点v，v是u的子结点。是不是很像背包问题？只是不同的是，这里的方程少了一维，原先应该是以i为根结点的子树中，从前j颗子树中选k个叶子结点所剩金钱的最大值，而实际上可以用滚动数组来省去j这一维。详见代码。

#include<cstdio>
#include<cstring>
#include<string>
#include<cctype>
#include<iostream>
#include<set>
#include<map>
#include<vector>
#include<stack>
#include<cmath>
#include<queue>
#include<algorithm>
#define lson l,m,rt<<1
#define rson m+1,r,rt<<1|1
typedef long long ll;
using namespace std;
const int maxn = 3000 + 5;
const int INF = 0x3f3f3f3f;
struct Edge {
  int to, cost;
  Edge(int v, int c) : to(v), cost(c) {}
};
vector<Edge> G[maxn];
int dp[maxn][maxn];
int cnt[maxn]; //结点观众的数量

void dfs(int u) {
  if(!G[u].size()) {
    cnt[u] = 1;
    dp[u][0] = 0;
    return;
  }
  dp[u][0] = 0;
  for(int i = 0; i < G[u].size(); i++) {
    int v = G[u][i].to, c = G[u][i].cost;
    dfs(v); cnt[u] += cnt[v];
    for(int j = cnt[u]; j >= 1; j--) { //j这一维是滚动数组，所以要逆向循环
      for(int k = 0; k <= j; k++) if(j-k <= cnt[v])
        dp[u][j] = max(dp[u][j], dp[u][k] + dp[v][j-k] - (j-k == 0 ? 0 : c));
    }
  }
}

int main() {
  int n, m;
  scanf("%d%d", &n, &m);
  for(int u = 1; u <= n-m; u++) {
    int x; scanf("%d", &x);
    while(x--) {
      int v, c; scanf("%d%d", &v, &c);
      G[u].push_back(Edge(v, c));
    }
  }
  memset(dp, -INF, sizeof dp);
  for(int i = n-m+1; i <= n; i++) scanf("%d", &dp[i][1]);
  dfs(1);
  for(int i = n; i >= 0; i--) {
    if(dp[1][i] >= 0) {
      printf("%d\n", i);
      break;
    }
  }
  return 0;
}