hdu 5534 Partial Tree【完全背包】

题意:给n-1个点,每个点有个权值和这个点连接的边的数量(从1到n-1)现在从里面选取n个点(可以多次选取)构成一颗树,求出最大权值。

例如:

4

5 1 4

第一条边权值为5,连接的边为1

第二条边权值为1,连接的边为2

第三条边权值为4,连接的边为3

权值最大情况为权值为4的点连接3个权值为5的点。

思路:对于任何一棵树,他的总边数是确定的,即边数=n-1,那么想构成一棵树,需要满足选取的点个数i=n,选取点的degree总和=2*(n-2)。

很容易想到一个完全背包,i表示选取了i个点,j表示已经选取点的degree总和,那么dp[i][j]=max(dp[i][j],dp[i-1][j-v[k]]+w[k])

for (int i = 1; i <= n; ++i) {
            for (int k = 0; k < n-1; ++k) {
                for (int j = a[k].d; j <= v; ++j) {
                    dp[i][j]=max(dp[i][j],dp[i-1][j-a[k].d]+a[k].w);
                }
            }
        }

想法很好,但是不可行。2000*2000*4000*t的复杂度显然会超时。

那么我们需要优化一下,三个循环必会T,只有优化到两重循环,那么考虑一下i和j的关系,只有将i和j合并到一起,才能实现两重循环。

i表示选取了i个点,j表示degree总和,i最终应该等于n,j最终应该等于2*n-2,容易看出,每选取一个新的节点,i的增量固定为1,而j的增量是不确定的,那么我先选n个degree=1的节点,此时j=n,距离目标2*n-2还差n-2,因为n个节点已经全部选了,只用考虑将degree增加到2*n-2即可,总增量=n-2。

预处理每个节点对于第一个节点的权值增量,degree增量,问题变成简单的完全背包。

#include"bits/stdc++.h"
using namespace std;
int dp[2020];
const int inf=0x3f3f3f3f;
struct node
{
    int d,w;
}a[2020];
int main()
{
    int t;
    scanf("%d",&t);
    while(t--)
    {
        int n;
        memset(dp,-inf,sizeof(dp));
        scanf("%d",&n);
        for (int i = 0; i < n-1; ++i) {
            scanf("%d",&a[i].w);
            a[i].d=i;
        }
        dp[0]=n*a[0].w;
        for (int i = 1; i < n-1; ++i) {
            a[i].w-=a[0].w;
        }
        for (int i = 1; i < n-1; ++i) {
            for (int j = a[i].d; j <= n-2; ++j) {
                dp[j]=max(dp[j],dp[j-a[i].d]+a[i].w);
            }
        }
        printf("%d\n",dp[n-2]);
    }
}

猜你喜欢

转载自blog.csdn.net/qq_42671946/article/details/83413764