【题解】洛谷 P1273 有线电视网 （树形dp）

这个题可以算是poj2486的简化版，但仍有许多细节要注意。

题目让求不赔本的情况下连接最多的用户数，又是树形结构，所以我们可以联想到树形dp。设二维数组dp[n][m]代表以n结点为父亲结点，连接m个用户时的收益。由于要求dp最大值，所以我们初始dp数组[n][m]为负无穷（后面要比较赔不赔本）。利用邻接表存图，读入数据。设dfs（x，fa）表示从x节点开始，父亲节点为fa♂。为了节约时间，又知道用户的标号肯定是n-m+1到n（题目说了），所以我们设size数组记录每个节点他子树中有多少个用户，若x>n-m代表其为用户端的结点，那么初始化dp[x][1]=money[x-(n-m)]，size[x]=1。枚举x结点相邻的所有节点，若不为父节点dfs下去，并让size[x]+=size[now]。做分组背包枚举用户数j从size[x]到0（节省时间），枚举x的子节点now要连k个用户（k<=j并且k不比size[now]大），然后可得状态转移方程dp[x][j]=max(dp[x][j],dp[x][j-k]+dp[now][k]-length[i]) 因为是求收益所以最后要减去连接x和now的边（也就是接线费用）。得到了dp数组后我们在主函数里枚举j从m到1，若dp[1][j]的值不是负数就代表电视台不赔钱，这时输出j即为能连接的最多用户数，然后终止程序即可。

#include<cstdio>
#include<algorithm>
#include<cstring>
#include<cstdlib>
#include<iostream>
#include<iomanip>
using namespace std;
const int maxn=3010;
const int maxm=6020;
int head[maxn],nnext[maxm],to[maxm],length[maxm];
int tot;
int money[maxm];
int n,m;
int dp[maxn][maxm];
int size[maxn];
void add(int x,int y,int l)
{
	tot++;
	nnext[tot]=head[x];
	head[x]=tot;
	to[tot]=y;
	length[tot]=l;
}
void dfs(int x,int fa)
{
	if(x>n-m)
	{
		dp[x][1]=money[x-(n-m)];
		size[x]=1;
		return ;
	}
	for(int i=head[x];i;i=nnext[i])
	{
		int now=to[i];
		if(now==fa) continue;
		dfs(now,x);
		size[x]+=size[now];	
		for(int j=size[x];j>=0;j--)
			for(int k=1;j-k>=0&&k<=size[now];k++)
				dp[x][j]=max(dp[x][j],dp[x][j-k]+dp[now][k]-length[i]);
	}
} 
int main()
{
	cin>>n>>m;
	for(int i=1;i<=n;i++)
	{
		for(int j=1;j<=m;j++)
		{
			dp[i][j]=-1e9;
		}
	}
	for(int i=1;i<=n-m;i++)
	{
		int k;
		cin>>k;
		for(int j=1;j<=k;j++)
		{
			int num,val;
			cin>>num>>val;
			add(i,num,val);
			add(num,i,val);
		}
	}
	for(int i=1;i<=m;i++)
	{
		cin>>money[i];
	}
	dfs(1,0);
	for(int j=m;j>=1;j--)
	{
		if(dp[1][j]>=0)
		{
			cout<<j<<endl;
			exit(0);
		}
	}
	return 0;
}