HihoCoder - 1104 Suzhou Adventure （树上背包）

题目：https://vjudge.net/contest/323605#problem/D

题意：给你一棵n个点的树，1点出发，然后规定k个点必须去，每个点上有一个权值，要走m个点，问最大权值是多少

思路：首先k个点因为是必须去的，所以我们先树形DP预处理求出因为这些必须要去的点会影响到其他几个点也必须去，然后我们再树上背包，去考虑m-k个点的最大值，如果当前值之前预处理过了，我们就不加上当前点的权值

#include<bits/stdc++.h>
#define maxn 105
#define mod 1000000007
using namespace std;
typedef long long ll;
int n,m,k;
int dp[maxn][maxn];
int d[maxn];
vector<int> mp[maxn];
int vis[maxn];
ll num;
void dfs(int x,int f){
    for(int i=;i<mp[x].size();i++){
        int u=mp[x][i];
        if(u==f) continue;
        dfs(u,x);
        for(int t=m;t>=;t--){
            for(int j=t;j>=;j--){
                if(t-j>=){
                    dp[x][t]=max(dp[x][t],dp[x][t-j]+dp[u][j]);
                }
            }
        }
    }
    if(vis[x]==){
        if(x!=){
            for(int t=m;t>;t--){
                dp[x][t]=dp[x][t-]+d[x];
            }
        }
    }
}
int nxt(int x,int f){
    int mx=;
    for(int i=;i<mp[x].size();i++){
        int u=mp[x][i];
        if(u==f) continue;
        int e=nxt(u,x);
        mx=max(mx,e);
    }
    if(vis[x]||mx){
        vis[x]=;
        num+=d[x];
        m--;
        return ;
    }
}
int main(){
    scanf("%d%d%d",&n,&k,&m);
    int x,y;
    for(int i=;i<=n;i++) scanf("%d",&d[i]);
    for(int i=;i<k;i++){
        scanf("%d",&x);
        vis[x]=;
    }
    for(int i=;i<n-;i++){
        scanf("%d%d",&x,&y);
        mp[x].push_back(y);
        mp[y].push_back(x);
    }
    nxt(,-);
    if(m<) printf("-1");
    else if(m==){
        printf("%lld",num);
    }
    else{
        dfs(,-);
        cout<<dp[][m]+num;
    }
}