【树形DP】 POJ 1155 / 1947 树形背包

题目链接: POJ 1155 TELE

分析:  用dp[i][j]表示在结点i下最j个用户公司的收益, 做为背包处理.

       dp[cnt][i+j] = max( dp[cnt][i+j] , dp[cnt][i]+dp[son][j]-pay );

      其中pay是cnt->son这一路径的成本

代码: 

#include <iostream>
#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <algorithm>
using namespace std;
const int inf = 0x7FFFFFFF;
const int maxn = 3005;

struct node{
    int to;
    int pay;
    node* next;
}tree[maxn],*head[maxn];

int w[maxn],num[maxn],tmp[maxn];
int dp[maxn][maxn];
int n,m,ptr;

void Init(){ 
    ptr=1;
    memset(num,0,sizeof(num));
    for(int i=1;i<=n;++i)
        for(int j=1;j<=m;++j)
            dp[i][j]=-inf;
}

void AddEdge(int a,int b,int c){
    tree[ptr].to=b;
    tree[ptr].pay=c;
    tree[ptr].next=head[a];
    head[a]=&tree[ptr++];
}

void DFS(int cnt){
    dp[cnt][0]=0;
    if(cnt>n-m){
        num[cnt]=1;
        dp[cnt][1]=w[cnt];
        return ;
    }
    node* p=head[cnt];
    while(p!=NULL){ 
        DFS(p->to);
        for(int i=0;i<=num[cnt];++i) 
            tmp[i]=dp[cnt][i]; ///不能放到下面的循环里, 因为可能有些会改变
        for(int i=0;i<=num[cnt];++i)
            for(int j=1;j<=num[p->to];++j)
                dp[cnt][i+j]=max(dp[cnt][i+j],tmp[i]+dp[p->to][j]-p->pay);
        num[cnt]+=num[p->to];  ///cnt的子结点最多能选用户的数量
        p=p->next;
    }
}

int main(){
    while(~scanf("%d%d",&n,&m)){
        Init();
        for(int i=1;i<=n-m;++i){
            int t; scanf("%d",&t);
            while(t--){
                int a,b; scanf("%d%d",&a,&b);
                AddEdge(i,a,b);
            }
        }
        for(int i=n-m+1;i<=n;++i)
            scanf("%d",w+i);
        DFS(1); 
        for(int i=m; i>=0; --i)
            if(dp[1][i]>=0) {
                printf("%d\n",i); break;
            }
    }
    return 0;
}

题目链接;  POJ 1947 Rebuilding Roads

题意: 从一棵树中分离出结点数为p的子树最少要剪去几条边.

分析; dp[i][j]表示从i结点中分离出j个结点的子树最少剪去的边数

     方程: dp[cnt][j+k]=min(dp[cnt][j+k],dp[cnt][j]+dp[son][k]-1)

代码:

///题意:  有一棵树n个结点, 问从中分离出一个结点数为m的树枝 最少要去掉几条边

#include<iostream>
#include<cstring>
#include<cstdio>
#include<vector>

using namespace std;
const int maxn=155;
const int inf=1000000;
int dp[maxn][maxn];
vector<int>Tree[maxn];
int n,p;
void Init(){
    for(int i=1;i<maxn;++i)
        for(int j=0;j<maxn;++j)
            dp[i][j]=inf;
}
void DFS(int cnt){
    int len=Tree[cnt].size();
    for(int i=0;i<len;++i)
        DFS(Tree[cnt][i]);
    dp[cnt][1]=len;
    for(int i=0;i<len;++i){
        int son=Tree[cnt][i];
        for(int j=p;j>=1;--j)
            for(int k=p-j;k>=1;--k)
                dp[cnt][j+k]=min(dp[cnt][j+k],dp[cnt][j]+dp[son][k]-1);
    }
}
int main(){
    Init();
    scanf("%d%d",&n,&p);
    for(int i=1;i<n;++i){
        int a,b; scanf("%d%d",&a,&b);
        Tree[a].push_back(b);
    }
    DFS(1);
    int ans=dp[1][p];
    for(int i=2;i<=n;++i)
        ans=min(dp[i][p]+1,ans);
    printf("%d\n",ans);
    return 0;
}