HDU - 5534 Partial Tree(dp)

题意：给你n个节点，连成一棵树，如果节点的度为i，则该节点的价值为f【i】。求树的最大价值。

思路：首先该树的总度数为2*（n-1），所以刚开始想到了二维dp，dp[i][j]为将j个度分给i个节点。最后dp[n][2*n-2]为答案，但是这样的时间复杂度为n的立方（超时）。看了题解才知道既然每个节点的度数最小是1，所以不妨先给每个节点分配一个节点，题目就变成了将n-2随意分配。就是算法导论的钢条切割问题了。如果将最后的答案加上n*f[1]就可以处理dp的时候节点分配到的度为0的情况，只要将每个节点的价值变成f[i+1] - f[1]就行。

#include <bits/stdc++.h>
using namespace std;
const int maxn = 5e3;
int dp[maxn], f[maxn];
int main()
{
//    freopen("in.txt", "r", stdin);
    int T, n;
    cin >> T;
    while(T--)
    {
        memset(dp, -0x3f, sizeof(dp));
        cin >> n;
        for(int i = 1; i < n; i++)
            cin >> f[i];
        dp[0] = 0;
        for(int i = 1; i <= n - 2; i++)
        {
            for(int j = 1; j <= i; j++)
                dp[i] = max(dp[i], dp[i - j] + f[j + 1] - f[1]);
        }
        cout << dp[n - 2] + n*f[1] << endl;
    }
    return 0;
}