当前位置:编程学习 > C/C++ >>

hdu 1561 the more,the better

第一个树形DP。这个题既是dp又是dfs和背包。dp[i][j]表示当前访问节点序号是i,且访问j次其子节点。dp[i][j]=max(dp[i][j],dp[i][j-r]+dp[t][r])。t为i的一个子节点。详细的看代码。

[cpp]
#include<iostream>  
#include<vector>  
using namespace std; 
vector<int> tree[201]; 
int dp[201][201]; 
//int val[201];  
int visited[201]; 
int n,m; 
int max(int a,int b) 

    if(a>b) 
        return a; 
    else return b; 

int main() 

    int i,father,v; 
    void dfs(int k); 
    while(cin>>n>>m) 
    { 
        if(n+m==0) 
            break; 
        for(i=0;i<=n;i++) 
            tree[i].clear(); 
        memset(dp,-1,sizeof(dp)); 
        memset(visited,0,sizeof(visited)); 
        for(i=1;i<=n;i++) 
        { 
            cin>>father>>v; 
            tree[father].push_back(i); 
            /*val[i]=v;*/ 
            dp[i][1]=v; 
        } 
        for(i=0;i<=n;i++) 
            dp[i][0]=0; 
        dp[0][1]=0; 
        m=m+1; 
        dfs(0); 
        cout<<dp[0][m]<<endl; 
    } 
    return 0; 

void dfs(int k) 

    int t,j,r,i; 
    visited[k]=1; 
    for(i=0;i<tree[k].size();i++) 
    { 
        t=tree[k][i]; 
        if(visited[t]==0) 
            dfs(t); 
        for(j=m;j>=1;j--) 
            for(r=1;r<j;r++) 
            { 
                if(dp[k][j-r]!=-1&&dp[t][r]!=-1) 
                    dp[k][j]=max(dp[k][j],dp[k][j-r]+dp[t][r]); 
            } 
    } 

         

#include<iostream>
#include<vector>
using namespace std;
vector<int> tree[201];
int dp[201][201];
//int val[201];
int visited[201];
int n,m;
int max(int a,int b)
{
    if(a>b)
        return a;
    else return b;
}
int main()
{
    int i,father,v;
    void dfs(int k);
    while(cin>>n>>m)
    {
  if(n+m==0)
   break;
        for(i=0;i<=n;i++)
            tree[i].clear();
        memset(dp,-1,sizeof(dp));
        memset(visited,0,sizeof(visited));
        for(i=1;i<=n;i++)
        {
            cin>>father>>v;
            tree[father].push_back(i);
            /*val[i]=v;*/
            dp[i][1]=v;
        }
        for(i=0;i<=n;i++)
            dp[i][0]=0;
        dp[0][1]=0;
        m=m+1;
        dfs(0);
        cout<<dp[0][m]<<endl;
    }
    return 0;
}
void dfs(int k)
{
    int t,j,r,i;
    visited[k]=1;
    for(i=0;i<tree[k].size();i++)
    {
        t=tree[k][i];
        if(visited[t]==0)
            dfs(t);
        for(j=m;j>=1;j--)
            for(r=1;r<j;r++)
            {
                if(dp[k][j-r]!=-1&&dp[t][r]!=-1)
                    dp[k][j]=max(dp[k][j],dp[k][j-r]+dp[t][r]);
            }
    }
}
       

 

 

补充:软件开发 , C++ ,
CopyRight © 2022 站长资源库 编程知识问答 zzzyk.com All Rights Reserved
部分文章来自网络,