poj1155 （树形dp）

最新推荐文章于 2019-08-20 15:09:37 发布

原创最新推荐文章于 2019-08-20 15:09:37 发布 · 687 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#tree #struct #c

dp 专栏收录该内容

14 篇文章

订阅专栏

本文介绍了一种结合图论和动态规划解决特定问题的算法实现。通过定义节点和边来构建图，并利用动态规划求解最优解。该算法适用于解决具有子问题重叠性质的问题。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

#include<stdio.h>
#include<string.h>
#define inf 1000000000
#define max(x,y) ((x)>(y)?(x):(y))
int N,M;
typedef struct{int  d,v,next;}node;
node tree[6010];
int vc[3001],value[3001];
int dp[3001][3001]; int ptr=1; int num[3001]={0};

void addedge(int u,int v,int c)
{
    tree[ptr].d=c;
    tree[ptr].v=v; tree[ptr].next=vc[u];
    vc[u]=ptr++;
}

void dfs(int k)
{
    if(k>N-M)
    {
        num[k]=1;
        dp[k][1]=value[k];
        return;
    }
    dp[k][0]=0;
    int u,v,i,j;
    u=vc[k];
    while(u!=-1)
    {
        v=tree[u].v;
        dfs(v);
        num[k]+=num[v];
        for(j=num[k];j>=0;j--)
            for(i=1;i<=num[v] && i<=j;i++)
                dp[k][j]=max(dp[k][j],dp[k][j-i]+dp[v][i]-tree[u].d);
        u=tree[u].next;
    }
}
int main()
{
    int i,j,k,a,c,h;
    memset(vc,-1,sizeof(vc)); memset(value,0,sizeof(value));
    //memset(T,0,sizeof(T));
    scanf("%d%d",&N,&M);
    for(i=1;i<=N-M;i++)
    {
        scanf("%d",&k);
        for(h=1;h<=k;h++)
        {
            scanf("%d%d",&a,&c);
           // T[i][a]=T[a][i]=c;
            addedge(i,a,c);
        }
    }
    for(i=N-M+1;i<=N;i++)
    {
        scanf("%d",&h); value[i]=h;
    }
    for(i=1;i<=N;i++)
        for(j=1;j<=N;j++)
            dp[i][j]=-inf;
    dfs(1);
    for(i=M;i>=0;i--)
    {
        if(dp[1][i]>=0)
        {
            printf("%d",i);
            break;
        }
    }
}