POJ 1155 TELE(树型DP)

27 篇文章 0 订阅

题意:有一个电视台广播节目,广播的网络用一棵树表示,节点1表示电台,叶子结点表示用户,用户愿意付一定的钱去收看这个节目,从非叶子结点到其他结点需要一定的费用(即从中继点到另一个中继点需要一些钱),问最后在不亏本的情况下,最多能使多少人收看到节目。

定义dp[i][j]表示在节点i为根节点的子树下,有j个人收看节点的最大利润。

状态转移方程是:dp[i][j]=max(dp[i][j],dp[i][j-k]+dp[son of i][k])

#include <iostream>
#include <cstdio>
#include <cstring>
using namespace std;
#define INF (1<<30)
const int N=3005;
struct Edge
{
	int v,w;
	Edge* nxt;
}memo[N*N],*cur,*adj[N];

int dp[N][N],n,m;
int num[N];

void addEdge(int u,int v,int w)
{
	cur->v=v; cur->w=w;
	cur->nxt=adj[u];
	adj[u]=cur++;
}
void dfs(int u,int fa)
{
	for(Edge* it=adj[u];it;it=it->nxt)
	{
		int v=it->v,w=it->w;
		if(v==fa) continue;
		dfs(v,u);
		num[u]+=num[v];
		for(int i=num[u];i>=0;i--)
			for(int j=0;j<=i;j++)
				if(dp[u][i-j]!=-INF&&dp[v][j]!=-INF) 
					dp[u][i]=max(dp[u][i],dp[u][i-j]+dp[v][j]-w);
	}
}
void init()
{
	cur=memo;
	memset(adj,0,sizeof(adj));
	memset(num,0,sizeof(num));
	for(int i=0;i<=n;i++)
	{
		dp[i][0]=0;
		for(int j=1;j<=m;j++) dp[i][j]=-INF;
	}
}
int main()
{
	while(scanf("%d%d",&n,&m)!=EOF)
	{
		init();

		int v,w,k;
		for(int i=1;i<=n-m;i++)
		{
			scanf("%d",&k);
			for(int j=0;j<k;j++)
			{
				scanf("%d%d",&v,&w);
				addEdge(i,v,w);
				addEdge(v,i,w);
			}
		}
		for(int i=n-m+1;i<=n;i++) 
			scanf("%d",&dp[i][1]),num[i]=1;

		dfs(1,-1);
		for(int i=m;i>=0;i--)
		{
			if(dp[1][i]>=0)
			{
				printf("%d\n",i);
				break;
			}
		}
	}
	return 0;
}


评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值