hdu5534 Partial Tree

题目地址：http://acm.hdu.edu.cn/showproblem.php?pid=5534

题目大致意思是有一棵树，n个点，n-1条边，对于树中的一个点有x条边连着，度数为x。对于度数为x的点的coolness为f(x)，输入给出，现要求构造一棵树，使得一棵树的coolness之和最大。

这题是一题动态规划题，是一个背包问题。稍微分析一下，n-1条边必然有2*(n-1)个度，现就需要将2*(n-1)个度分配到n个点上，然后可以做DP，将2*(n-1)个物品分配到n个箱子上，每个箱子可以放1~x个物品，但是这样的话时间效率是O(n^3)，会超时。所以要加点优化，由于这题目有限制，是一棵树，所以每个点必然至少有一个度，所以之后就只需要将n-2个物品分配到n个箱子上，这样就没有n个箱子的限制了，效率就变成了O(n^2)。同时对数据进行一下处理，对f(x)做关于f(1)的相对值，然后DP即可。

代码附上：

#include<iostream>
#include<cstdio>
#include<cmath>
using namespace std;
int T,i,s,j,n,a[12020],f[12020];
int max(int a,int b)
{
    if (a>b) return a; else return b;
}
int main()
{
  scanf("%d",&T);
  while (T--)
  {
      scanf("%d",&n);
      for (i=1;i<=n-1;i++)
      {
          scanf("%d",&a[i]);
          if (i!=1) a[i]-=a[1];
      }
      s=a[1]*n;
      for (i=1;i<=n-2;i++) f[i]=-20160000;
      f[0]=0;
      for (i=2;i<=n-1;i++)
        for (j=i-1;j<=n-2;j++)
            f[j]=max(f[j],f[j-i+1]+a[i]);
       printf("%d\n",s+f[n-2]);
  }
  return 0;
}