poj1155 TELE(树形+背包dp)

题意:某电台要广播一场比赛,该电台网络是由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;
           }
        }
    }
}
  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值