POJ1155 TELE(树DP)-CSDN博客

本文链接：https://blog.csdn.net/kopyh/article/details/52948621

该博客介绍了如何利用树形动态规划（Tree DP）解决POJ1155问题。博主详细阐述了题意，即在传输给j个用户的情况下，如何使第i个节点的花费最小，并给出了利用dp状态转移的思想和关键代码片段。重点在于处理节点间的最大用户数，以及在枚举过程中避免重复更新。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

题目：

poj.org/problem?id=1155

题意：

有N个节点的一棵树，上面前n-m个节点是信号站，后m个节点是用户，每个节点之间传输有消耗，每个用户有费用，总信号从1节点传输出来，求在不赔本的情况下能够最多传输给多少个用户。

思路：

明显是树DP，考虑3000个节点，dp[i][j]就是第i个节点在传输给j个用户的情况下的最小花费是多少，记录每个节点取得的可能最大用户数目，然后对于每个子节点暴力枚举更新就行了，最后1节点时花费为非正的最大的用户数就是答案。

注意每次更新一个节点的子节点时要把当前DP值储存下来用于更新，否则枚举不同数目时可能会重复。

代码：

//kopyh
#include <bits/stdc++.h>
#define INF 0x3f3f3f3f
#define MOD 1000000007
#define N 3001
using namespace std;
int n,m,sum,res,flag;
int dp[N][N],val[N],num[N],vis[N],tmp[N];
vector<int>g[N],f[N];
void dfs(int now)
{
    dp[now][0] = 0;
    if(val[now]>=0)dp[now][1] = -val[now], num[now]=1;
    for(int i=0;i<g[now].size();i++)
    {
        int y=g[now][i];
        dfs(y);
        for(int j=0;j<=num[now];j++)tmp[j]=dp[now][j];
        for(int j=0;j<=num[now];j++)
            for(int k=0;k<=num[y];k++)
                dp[now][j+k] = min(dp[now][j+k],tmp[j]+dp[y][k]+f[now][i]);
        num[now]+=num[y];
    }
}
int main()
{
    int i,j,k,cas,T,t,x,y,z;
    while(scanf("%d%d",&n,&m)!=EOF)
    {
        for(i=1;i<=n;i++)g[i].clear(),f[i].clear();
        for(i=1;i<=n-m;i++)
        {
            scanf("%d",&t);
            while(t--)
            {
                scanf("%d%d",&x,&y);
                g[i].push_back(x);
                f[i].push_back(y);
            }
        }
        memset(val,-1,sizeof(val));
        memset(num,0,sizeof(num));
        memset(dp,0x3f,sizeof(dp));
        for(i=n-m+1;i<=n;i++)
            scanf("%d",&val[i]);
        dfs(1);
        for(i=m;i>=0;i--)
            if(dp[1][i]<=0)
                break;
        printf("%d\n",i);
    }
    return 0;
}