题意:某电台要广播一场比赛,该电台网络是由N个网点组成的一棵树,其中M个点为客户端,其余点为转发站。客户端i愿支付的钱为pay[i],每一条边需要的花费固定,问电台在保证不亏损的情况下,最多能使多少个客户端接收到信息?
思路:
设dp[i][j]:表示i节点为根节点并且子树已经有j个用户的时候最大剩余费用。
那么人人为我的转移方程式的写法就是:
dp[i][j]=max(dp[i][j-k]+dp[v][k]); (v是i的子节点)。 (1<=k<=num[v])
注意:
1.num[v]表示v为根的子树里叶子节点即客户的个数,而非子节点的个数。
2.dp[][]转态可能为负,所以初始化为负无穷。
然后时间复杂度我个人觉得好像是O(n^3)? 为啥没TLE? 我也不造。。。。
#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
template<int N,int M>//N点的个数,M边的个数
struct Graph
{
int top;
struct Vertex{
int head;
}V[N];
struct Edge{
int v,next,w;
}E[M];
void init(){
memset(V,-1,sizeof(V));
top = 0;
}
void add_edge(int u,int v,int w){
E[top].v = v;
E[top].next = V[u].head;
E[top].w=w;
V[u].head = top++;
}
};
const int N=3000+10,inf=1e9;
Graph<N,N*2> g;
int m,n,dp[N][N],num[N],val[N];
void dfs(int u,int f){
for(int i=g.V[u].head;i!=-1;i=g.E[i].next){
int v=g.E[i].v,w=g.E[i].w;
if(v==f) continue;
dfs(v,u);num[u]+=num[v];
for(int j=num[u];j>=1;j--){
for(int k=1;k<=num[v];k++){
dp[u][j]=max(dp[u][j],dp[u][j-k]+dp[v][k]-w);
}
}
}
}
int main(){
while(~scanf("%d%d",&n,&m)){
g.init();
for(int i=1;i<=n-m;i++){
int k;scanf("%d",&k);
for(int j=0;j<k;j++){
int a,b;scanf("%d%d",&a,&b);
g.add_edge(i,a,b);
g.add_edge(a,i,b);
}
}
for(int i=1;i<n;i++){
for(int j=1;j<=m;j++) dp[i][j]=-inf;
}
memset(num,0,sizeof(num));
for(int i=n-m+1;i<=n;i++) { scanf("%d",&dp[i][1]);num[i]=1;}
dfs(1,0);
for(int i=m;i>=1;i--){
if(dp[1][i]>=0){
printf("%d\n",i);
break;
}
}
}
}