poj 1155 TELE （树形dp）

题目链接：http://poj.org/problem?id=1155
参考了这位acmer的，我用的是vector的方法：http://blog.csdn.net/woshi250hua/article/details/7635680
dp[i][j]:以结点i为根节点时拥有j个客户的最大价值。

• 当结点i为叶子结点时，dp[i][1]=money[i];
• 当结点i非叶子结点，dp[i][j]=max(dp[i][j],dp[i][j-k]+dp[d][k]-len[d]);
  其中money[i]为某用户愿意付出的价值。d为i的一个子结点，k代表在d结点中选k个客户。

#include<iostream>
#include<cstdio>
#include<vector>
using namespace std;
const int INF=1<<30;
const int N=3005;
vector<int> a[N];
int dp[N][N];//dp[i][j]表示以i为根节点，有j个用户产生的最大价值
int sum[N]; //sum[i]表示结点i有多少个子结点。
int money[N],len[N];
int n,m;
void init()
{
    memset(sum,0,sizeof(sum));
    for(int i=0;i<=n;i++)
        for(int j=0;j<=n;j++)
            dp[i][j]=-INF;
}
void DP(int v)
{
    if(v>=n-m+1&&v<=n)
    {
        dp[v][1]=money[v];
        sum[v]=1;
        return ;
    }
    dp[v][0]=0;
    for(int i=0;i<a[v].size();i++)
    {
        int d=a[v][i];
        DP(d);
        sum[v]+=sum[d];
        for(int j=sum[v];j>0;j--)//从大到小更新，以保证数据不会出现干扰。
        {
            for(int k=0;k<=sum[d];k++)
            {
                if(j>=k&&dp[d][k]!=-INF&&dp[v][j-k]!=-INF)
                    dp[v][j]=max(dp[v][j],dp[d][k]+dp[v][j-k]-len[d]);
            }
        }
    }
    return;
}
int main()
{
    //freopen("in.txt","r",stdin);
    while(cin>>n>>m)
    {
        for(int i=1;i<=n-m;i++)
        {
            int t;
            scanf("%d",&t);
            int b,c;
            for(int j=1;j<=t;j++)
            {
                cin>>b>>c;
                len[b]=c;
                a[i].push_back(b);
            }
        }
        for(int i=n-m+1;i<=n;i++)
            scanf("%d",&money[i]);
        init();
        DP(1);
        for(int i=m;i>=0;i--)
        {
            if(dp[1][i]>=0)
            {
                cout<<i<<endl;
                break;
            }
        }
    }
    return 0;
}

又试了试其他方法，其实基本思路是差不多的，只是实现方法有些许区别。

#include<iostream>
#include<cstdio>
#include<algorithm>
using namespace std;
const int N=3005;
const int INF=1<<30;
struct node
{
    int v,len;
    node* next;
}tree[N*2],*head[N*2];
int n,m;
int sum[N],dp[N][N];
int money[N],len[N];
int vis[N];
int tmp;
void init()
{
    memset(head,NULL,sizeof(head));
    memset(sum,NULL,sizeof(sum));
    memset(vis,NULL,sizeof(vis));
    tmp=0;
}
void AddEdge(int u,int v,int len)
{
    tree[tmp].v=v; tree[tmp].len=len;
    tree[tmp].next=head[u]; head[u]=&tree[tmp++];
    tree[tmp].v=u; tree[tmp].len=len;
    tree[tmp].next=head[v]; head[v]=&tree[tmp++];
}
void dfs(int num)
{
    dp[num][0]=0;
    if(num>=n-m+1&&num<=n)
    {
        sum[num]=1;
        dp[num][1]=money[num];
        return;
    }
    vis[num]=1;
    node* p=head[num];
    while(p!=NULL)
    {
        if(!vis[p->v])
        {
            dfs(p->v);
            sum[num]+=sum[p->v];
        }
        p=p->next;
    }
    p=head[num];
    while(p!=NULL)
    {
        int k=p->v;
        int Len=p->len;
        for(int i=sum[num];i>=0;i--)
        {
            for(int j=0;j<=sum[k];j++)
            {
                if(i>=j&&dp[num][i-j] != -INF && dp[k][j] != -INF)
                    dp[num][i]=max(dp[num][i],dp[num][i-j]+dp[k][j]-Len);
            }
        }
        p=p->next;
    }
}
int main()
{
    //freopen("in.txt","r",stdin);
    while(cin>>n>>m)
    {
        init();
        int t,a,b;
        for(int i=1;i<=n-m;i++)
        {
            scanf("%d",&t);
            for(int j=1;j<=t;j++)
            {
                scanf("%d%d",&a,&b);
                len[a]=b;
                AddEdge(i,a,b);
            }
        }
        for(int i=n-m+1;i<=n;i++)
            scanf("%d",&money[i]);
        for(int i=0;i<=n;i++)
            for(int j=0;j<=n;j++)
                dp[i][j]=-INF;
        dfs(1);
        for(int i=m;i>=0;i--)
            if(dp[1][i]>=0)
            {
                cout<<i<<endl;
                break;
            }
    }
    return 0;
}