HDU 5534 Partial Tree(考虑树性质的dp)

$告诉你度数为d的点价值是f(d)，让你求一棵树，让他所有点的价值之和最大$
$开头考虑是一个背包，取n个东西，有n-1个东西，每个无限，价值f(i)$
$取2n-2的重量要求价值最大$
$然后复杂度是O(n^3)的，并且没有什么好方法优化$
$考虑树的性质，能否把去n件物品优化掉，然后就是常规的dp了$
$考虑树上点的度数至少是1，所以可以先给每个点附上度数1$
$然后还剩n-2的容量，其他点的度数都-1$
$然后就是让你选一些其他重量的东西，价值是原价值减去f[1]$
$然后这样取绝对不会超过n个$
$dp[i][j]=max(dp[i][j-(i-1)]+f[i],dp[i-1][j])$
$表示取度数i的点(但是当前的度数要-1，因为扣掉了原来的一点)$

---

代码：

#include <map>
#include <set>
#include <stack>
#include <queue>
#include <cmath>
#include <string>
#include <vector>
#include <cstdio>
#include <cctype>
#include <cstring>
#include <sstream>
#include <cstdlib>
#include <iostream>
#include <algorithm>
#pragma comment(linker,"/STACK:102400000,102400000")

using namespace std;
#define   MAX           100005
#define   MAXN          1000005
#define   maxnode       105
#define   sigma_size    30
#define   lson          l,m,rt<<1
#define   rson          m+1,r,rt<<1|1
#define   lrt           rt<<1
#define   rrt           rt<<1|1
#define   middle        int m=(r+l)>>1
#define   LL            long long
#define   ull           unsigned long long
#define   mem(x,v)      memset(x,v,sizeof(x))
#define   lowbit(x)     (x&-x)
#define   pii           pair<int,int>
#define   bits(a)       __builtin_popcount(a)
#define   mk            make_pair
#define   limit         10000

//const int    prime = 999983;
const int    INF   = 0x3f3f3f3f;
const LL     INFF  = 0x3f3f;
const double pi    = acos(-1.0);
const double inf   = 1e18;
const double eps   = 1e-8;
const LL     mod   = 1e9+7;
const ull    mx    = 133333331;

/*****************************************************/
inline void RI(int &x) {
      char c;
      while((c=getchar())<'0' || c>'9');
      x=c-'0';
      while((c=getchar())>='0' && c<='9') x=(x<<3)+(x<<1)+c-'0';
 }
/*****************************************************/

int dp[2][2020];
int a[2020];
int main(){
    int t;
    cin>>t;
    while(t--){
        int n;
        scanf("%d",&n);
        for(int i=1;i<n;i++){
            scanf("%d",&a[i]);
            if(i!=1) a[i]-=a[1];
        }
        mem(dp,0);
        dp[1][0]=n*a[1];
        int cnt=1;
        for(int i=2;i<n;i++){
            cnt^=1;
            for(int j=0;j<=n-2;j++){
                dp[cnt][j]=dp[cnt^1][j];
                if(j>=(i-1)) dp[cnt][j]=max(dp[cnt][j],dp[cnt][j-(i-1)]+a[i]);
            }
        }
        cout<<dp[cnt][n-2]<<endl;
    }
    return 0;
}