
本文共 2358 字,大约阅读时间需要 7 分钟。
题目
题目概要
给定一个正整数系数的 k k k 次函数 c ( x ) c(x) c(x) 表示度数为 x x x 的节点权值。求构建一棵树,使得权值和最大。
数据范围与约定
n ≤ 3000 , k ≤ 10 n\le 3000,k\le 10 n≤3000,k≤10 。
思路
考虑要构建一棵树, prufer \text{prufer} prufer 序列会是一个不错的选择。
根据 prufer \text{prufer} prufer 的性质,每个点出现的次数为度数减 1 1 1 ,并且总序列长度为 n − 2 n-2 n−2 。
发现权值与节点编号无关。于是可以当成做一个 完全背包(往 prufer \text{prufer} prufer 序列里赛点):度数为 x x x 的点,体积是 x − 1 x-1 x−1 ,权值是 c ( x ) c(x) c(x) 。
我们还得要求背包内物品的数量恰好为 n n n ——表示 n n n 个点的度数都被考虑了。
没有关系!我们有体积为 0 0 0 的物品,可以随便填。体积为 0 0 0 的不能用普通的背包进行转移,必须特殊考虑。
有一个方法:开始时把 n n n 个体积为 0 0 0 的物品放进背包,然后剩下的过程就是每次将若干个体积为 0 0 0 的物品替换为体积为 x x x 的物品。即 f 0 = n ⋅ c ( 1 ) f_0=n\cdot c(1) f0=n⋅c(1) 。
不仅解决了 0 0 0 的问题,顺便还保证了背包内恰好有 n n n 个物品。真好啊 😂
会不会替换过多?不会。因为体积不为 0 0 0 的物品,根本放不了 n n n 个。
那么 f x = max 2 ≤ j ≤ x + 1 [ f x − ( j − 1 ) + c ( j ) − c ( 1 ) ] f_x=\max_{2\le j\le x+1}{\left[f_{x-(j-1)}+c(j)-c(1)\right]} fx=2≤j≤x+1max[fx−(j−1)+c(j)−c(1)]
但是本题还要输出方案。这就要求 d p dp dp 的时候记录转移的前驱。利用这个我们可以计算出最终的树上的 n n n 个点在 prufer \text{prufer} prufer 序列上出现次数。
然后利用这个贪心的思想构建这棵树(度数越小的放末尾,事实证明,把所有叶子结点放在末尾即可),注意根的特殊性(对于非根节点,在 prufer \text{prufer} prufer 序列出现次数 = = = 儿子数;对于根,在 prufer \text{prufer} prufer 序列出现次数 = = = 儿子数 − 1 - 1 −1 )。
复杂度 O ( n 2 ) \mathcal O(n^2) O(n2) 。
#include <bits/stdc++.h>using namespace std;const int N = 3005, mod = 59393;int n, k, tot, a[N], c[N], f[N], g[N], d[N];void dfs (int x) { for (int i = 1; i <= d[x]; ++i) { printf("%d %d\n", x, ++tot); dfs(tot); }}int main () { scanf("%d%d", &n, &k); for (int i = 0; i <= k; ++i) { scanf("%d", &a[i]); } for (int i = 0; i <= n; ++i) { for (int j = k; ~j; --j) { c[i] = c[i] * i % mod + a[j]; c[i] %= mod; } } if (n == 1) { printf("0 %d\n", c[0]); return 0; } if (n == 2) { printf("1 %d\n",2 * c[1]); printf("1 2\n"); return 0; } f[0] = n * c[1]; for (int i = 2; i <= n; ++i) { for (int j = i - 1; j <= n - 2; ++j) { if (f[j] < f[j - (i - 1)] + c[i] - c[1]) { f[j] = f[j - (i - 1)] + c[i] - c[1]; g[j] = j - (i - 1); } } } printf("%d %d\n", n - 1, f[n - 2]); for (k = n - 2; k; k = g[k]) { d[++d[0]] = k - g[k]; } sort(d + 1, d + d[0] + 1); ++d[1]; dfs(tot = 1); return 0;}
鸣谢
感谢提供了思路和代码。果然我是个动态规划的矮子
发表评论
最新留言
关于作者
