[hihocoder1063]树形DP

题目描述

假设当前作战区域是一棵有根树，编织者所在的位置为根节点1，树中每个节点，有一个权值vi，代表这个节点的收益。树中的每条边，有一个权值wi，代表每条边的长度。编织者从根结点出发，最远累计移动d 距离时，所能得到的收益的最大值是多少？注意重复经过一个节点收益只能计算一次。

算法思路

我们很显然可以得到一种算法思路是:dp[i][j],表示从第i个结点走j个单位的路程可以得到的最大收益,但是这一题中的路程的长度长达$10^6$单位级别,这样做背包是会爆空间的。。
但是,我们再读一遍题目后发现,每个点的价值最多为2,也就是说总价值最大为200,所以dp[i][j]表示从第i个点出发收益为j所得到的最小路程.
然而，如果我们要访问一棵树的多个分支,则有的时候需要回朔，所以我们需要加入一个维度dp[i][j][k],表示从这个点出发返回或者不返回得到收益k的最短路程。
所以，我们就可以知道下面的转移关系:
1. 对于需要返回的情况，只有可能从各个子树都返回
2. 对于不需要返回的情况,只可以存在一颗子树不返回。
所以，转移方程是:

$$dp[cur][1][j]=max(2*len(cur,k)+dp[cur][1][j]+dp[k][1][j-t])$$
$$dp[cur][0][j]=(2*len(cur,k)+dp[cur][0][j]+dp[k][1][j-t]),len(cur,k)+dp[cur][1][j]+dp[k][0][j-t])$$
对于每一个查询，只需要遍历dp[1][0],找到第一个比查询值大的数即可。

代码

#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cstring>
#include<utility>
#include<vector>
using namespace std;

const int MAXN = 200;
const int MAXM = 105;
#define INF 0x3f3f3f3f

int dp[105][2][205];
int v[MAXM],n;
vector<pair<int,int> >grid[MAXM];

void Dfs(int cur,int fa)
{
    int i,j,k,t;

    for(i=0;i<=v[cur];i++)
        dp[cur][1][i] = dp[cur][0][i] = 0;

    for(i=0;i<grid[cur].size();i++){
        j = grid[cur][i].first;
        if(j!=fa){
            Dfs(j,cur);
            for(k=MAXN;k>=v[cur];k--){
                dp[cur][0][k] = min(dp[cur][0][k+1],dp[cur][0][k]);
                for(t=0;t<=k;t++){
                    dp[cur][0][k] = min(dp[cur][0][k],dp[cur][0][t]+2*grid[cur][i].second+dp[j][1][k-t]);
                    dp[cur][0][k] = min(dp[cur][0][k],dp[cur][1][t]+grid[cur][i].second+dp[j][0][k-t]);
                }
            }
            for(k=MAXN;k>=v[cur];k--){
                dp[cur][1][k] = min(dp[cur][1][k+1],dp[cur][1][k]);
                for(t=0;t<=k;t++)
                    dp[cur][1][k] = min(dp[cur][1][k],dp[cur][1][t]+dp[j][1][k-t]+2*grid[cur][i].second);
            }
        }
    }
}

int main()
{
    freopen("input","r",stdin);
    int i,j;
    int q;
    int tmp1,tmp2,tmp3;
    scanf("%d",&n);

    for(i=1;i<=n;i++)
        scanf("%d",&v[i]);

    for(i=0;i<n-1;i++){
        scanf("%d%d%d",&tmp1,&tmp2,&tmp3);
        grid[tmp1].push_back(make_pair(tmp2,tmp3));
        grid[tmp2].push_back(make_pair(tmp1,tmp3));
    }

    memset(dp,INF,sizeof(dp));


    Dfs(1,-1);
    scanf("%d",&q);
    for(i=0;i<q;i++){
        scanf("%d",&tmp1);
        //cout << tmp1 << endl;
        for(j=v[1];j<=MAXN;j++)
            if(dp[1][0][j]>tmp1)break;
        printf("%d\n",j-1);
    }
    return 0;
}