Partial Tree HDU5534(完全背包）

       Partial Tree HDU5534

---

题意

给一个n个节点的树，现上面没有边，需要你进行连边构造一棵树，使得其所有节点价值和最大。节点价值的定义与度有关，f[i]表示度为i的节点的价值。

输入

第一行输入一个整数t，表示样例组数；
然后输入n，表示树的节点个数；
然后一行紧跟n-1个数表示f[1]-f[n-1];

思路

这题推出一个常识 一棵n个节点的树 每个节点的度之和 = 2n-2
然后一颗树起码有两个节点的度为1 所以题目就转化为 用体积为 1 -> n-1 价值为f[i]的物品
要恰用n-2个物品装满容积为2n-4的背包使之容量最大
但是dp复杂度是n^3的 TLE
相当于多了一个限制 多了一维
于是想到 因为每个节点的度起码为1 所以可以先给每个节点分配1的度
于是取n个物品的限制就没有了 对于多下来的n-2个度 只要随便分配给n个节点即可
因为现在n个节点都已经分配到了1的度
所以对于体积为2->n-1的物品 现在将他们的价值都减去f[1] 相当于在原来的基础上加i-1的度
问题转化为一个完全背包问题 用2->n-2这些物品(但是体积都减1了)装下n-2的背包的最大大价值
一个坑点是 dp数组初值要赋成-INF 因为f[i]-f[1]可能是负数

代码

#include<cstdio>
#include<cstring>
#include<algorithm>
#include<vector>
#include<queue>
#include<cmath>
#include<map>
using namespace std;
#define INF 0x3f3f3f3f
#define maxn 2020
#define maxm 20005
#define ll long long int
int t,n,ans;
int f[maxn],dp[maxn];
int main()
{
    scanf("%d",&t);
    while(t--){
        ans=0;
        memset(dp,-INF,sizeof(dp));
        scanf("%d",&n);
        for(int i=1;i<n;i++)scanf("%d",&f[i]);
        ans+=n*f[1];
        for(int i=2;i<n;i++)f[i]-=f[1];
        dp[0]=0;
        for(int i=2;i<n;i++){
            for(int j=0;j+i-1<=n-2;j++){
                dp[j+i-1]=max(dp[j+i-1],dp[j]+f[i])；
            }
        }
        ans+=dp[n-2];
        printf("%d\n",ans);
    }
    return 0;
}