poj2486---Apple Tree(树状DP+背包)

https://vjudge.net/problem/POJ-2486

题意:一颗树,n个点(1-n),n-1条边,每个点上有一个权值,求从1出发,走V步,最多能遍历到的权值。

思路:考虑到能从父节点走到某一个子节点后,在走回至父节点,再取走其他的子节点(所以这个地方有3种情况).另开一维数组表示是否回到该结点。

dp[i][j][state]表示以i为父节点最多走j步,state表示状态,是否回到该节点.

dp数组初始化的时候,如果已经在i结点上了,那么他不管走几步起码都是weight[i];

https://blog.csdn.net/qq_38924883/article/details/82987165 有类似之处

#include<iostream>
#include<cstring>
#include<vector>
#define maxn 105
using namespace std;
vector<int>edge[maxn];
int dp[maxn][maxn<<1+5][2];
///dp[i][j][state]表示在子树i中最多走j步。
///state==0表示回到该点,否则不回.
int n,m;
int weight[maxn];
void dfs(int u,int father)
{
    for(int i=0;i<=m;i++)
        dp[u][i][0]=dp[u][i][1]=weight[u];
    for(int i=0;i<edge[u].size();i++)
    {
        int v=edge[u][i];
        if(v==father) continue;
        dfs(v,u);
        for(int j=m;j>=1;j--)///背包
        {
            for(int k=0;k<j;k++)///区间更新
            {
               ///1表示返回
                if(j-k>=2){
                dp[u][j][1]=max(dp[u][j][1],dp[v][k][1]+dp[u][j-k-2][1]);
                ///遍历完其他子树后回到u,在遍历完v的子树后返回v,返回u.
                  dp[u][j][0]=max(dp[u][j][0],dp[v][k][1]+dp[u][j-k-2][0]);
                ///遍历v的子树,返回v,返回u再去遍历其他子树
                }
                dp[u][j][0]=max(dp[u][j][0],dp[v][k][0]+dp[u][j-k-1][1]);
                ///遍历完其他子树后回到u,在遍历v的子树,不返回
            }
        }
    }
}
int main()
{
    while(cin>>n>>m)
    {
        memset(dp,0,sizeof(dp));
     ///   memset(weight,0,sizeof(weight));
        for(int i=1;i<=n;i++)
        {
            edge[i].clear();
            cin>>weight[i];
        }
        for(int i=1;i<n;i++)
        {
            int u,v;
            cin>>u>>v;
            edge[u].push_back(v);
            edge[v].push_back(u);
        }
        dfs(1,0);
       /// cout<<dp[1][m][0]<<endl;
       cout<<max(dp[1][m][0],dp[1][m][1])<<endl;
    }
    return 0;
}

猜你喜欢

转载自blog.csdn.net/qq_38924883/article/details/82987100