POJ 1155 TELE 树形dp

84 阅读 0 评论 56 点赞

我是靠谱客的博主含蓄手机，这篇文章主要介绍POJ 1155 TELE 树形dp，现在分享给大家，希望可以做个参考。

题目描述】
某收费有线电视网计划转播一场重要的足球比赛。他们的转播网和用户终端构成一棵树状结构，这棵树的根结点位于足球比赛的现场，树叶为各个用户终端，其他中转站为该树的内部结点。从转播站到转播站以及从转播站到所有用户终端的信号传输费用都是已知的，一场转播的总费用等于传输信号的费用总和。
现在每个用户都准备了一笔费用想观看这场精彩的足球比赛，有线电视网有权决定给哪些用户提供信号而不给哪些用户提供信号。
写一个程序找出一个方案使得有线电视网在不亏本的情况下使观看转播的用户尽可能多。

【输入数据】
第1行：2个整数N和M，其中2≤N≤3000，1≤M≤N-1，N为整个有线电视网的结点总数，M为用户终端的数量。第一个转播站即树的根结点编号为1，其他的转播站编号为2到N-M，用户终端编号为N-M+1到N。
接下来的N-M行每行表示—个转播站的数据，第i+1行表示第i个转播站的数据，其格式如下： K A1 C1 A2 C2 … Ak Ck K表示该转播站下接K个结点(转播站或用户)，每个结点对应一对整数A与C，A表示结点编号，C表示从当前转播站传输信号到结点A的费用。
最后一行依次表示所有用户为观看比赛而准备支付的钱数。

【输出数据】
仅一行，包含一个整数，表示上述问题所要求的最大用户数。

【样例输入】
5 3
2 2 2 5 3
2 3 2 4 3
3 4 2

【样例输出】
2

复制代码

#include<stdio.h>
#include<algorithm>
#include<cstring>
#define clear(a) memset(a,-1,sizeof(a));
using namespace std;
const int maxn=3005;
int dp[maxn][maxn],h[maxn],son[maxn],tem[maxn],n,m,a,b,num,k;
inline const int read(){
    register int f=1,x=0;
    register char ch=getchar();
    while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
    while(ch>='0'&&ch<='9'){x=(x<<3)+(x<<1)+ch-'0';ch=getchar();}
    return x*f;
}
struct data{
    int v,nxt,w;
}e[maxn];
inline void add(int u,int v,int w){
    num++;
    e[num].v=v;
    e[num].w=w;
    e[num].nxt=h[u];
    h[u]=num;
}
inline void dfs(int u){
    int v;
    for(register int i=h[u];i!=-1;i=e[i].nxt){
        v=e[i].v,dfs(v);
        for(register int j=0;j<=son[u];j++) tem[j]=dp[u][j];
        for(register int j=0;j<=son[u];j++)
            for(register int k=1;k<=son[v];k++)
                dp[u][k+j]=max(dp[u][k+j],tem[j]+dp[v][k]-e[i].w);
        son[u]+=son[v];
    }
}
int main(){
     n=read(),m=read();
     clear(h);
     for(register int i=1;i<=n-m;i++){
         k=read();
         for(register int j=1;j<=k;j++){
             a=read(),b=read();
             add(i,a,b);
         }
     }
     for(register int i=1;i<=n;i++)
         for(register int j=1;j<=m;j++) dp[i][j]=-100000;
     for(register int i=n-m+1;i<=n;i++) {son[i]=1;dp[i][1]=read();}
     dfs(1);   
     for(register int i=m;i>=0;i--)
         if(dp[1][i]>=0) {printf("%dn",i);break;}
}

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
#include<stdio.h>
#include<algorithm>
#include<cstring>
#define clear(a) memset(a,-1,sizeof(a));
using namespace std;
const int maxn=3005;
int dp[maxn][maxn],h[maxn],son[maxn],tem[maxn],n,m,a,b,num,k;
inline const int read(){
    register int f=1,x=0;
    register char ch=getchar();
    while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
    while(ch>='0'&&ch<='9'){x=(x<<3)+(x<<1)+ch-'0';ch=getchar();}
    return x*f;
}
struct data{
    int v,nxt,w;
}e[maxn];
inline void add(int u,int v,int w){
    num++;
    e[num].v=v;
    e[num].w=w;
    e[num].nxt=h[u];
    h[u]=num;
}
inline void dfs(int u){
    int v;
    for(register int i=h[u];i!=-1;i=e[i].nxt){
        v=e[i].v,dfs(v);
        for(register int j=0;j<=son[u];j++) tem[j]=dp[u][j];
        for(register int j=0;j<=son[u];j++)
            for(register int k=1;k<=son[v];k++)
                dp[u][k+j]=max(dp[u][k+j],tem[j]+dp[v][k]-e[i].w);
        son[u]+=son[v];
    }
}
int main(){
     n=read(),m=read();
     clear(h);
     for(register int i=1;i<=n-m;i++){
         k=read();
         for(register int j=1;j<=k;j++){
             a=read(),b=read();
             add(i,a,b);
         }
     }
     for(register int i=1;i<=n;i++)
         for(register int j=1;j<=m;j++) dp[i][j]=-100000;
     for(register int i=n-m+1;i<=n;i++) {son[i]=1;dp[i][1]=read();}
     dfs(1);   
     for(register int i=m;i>=0;i--)
         if(dp[1][i]>=0) {printf("%dn",i);break;}
}