【动态规划】【NOI2009】二叉查找树

【问题描述】
已知一棵特殊的二叉查找树。根据定义,该二叉查找树中每个结点的数据值都比它左子树结点的数据值大,而比它右子树结点的数据值小。
另一方面,这棵查找树中每个结点都有一个权值,每个结点的权值都比它的儿子结点的权值要小。
已知树中所有结点的数据值各不相同;所有结点的权值也各不相同。这时可得出这样一个有趣的结论:如果能够确定树中每个结点的数据值和权值,那么树的形态便可以唯一确定。因为这样的一棵树可以看成是按照权值从小到大顺序插入结点所得到的、按照数据值排序的二叉查找树。
一个结点在树中的深度定义为它到树根的距离加1。因此树的根结点的深度为1。
每个结点除了数据值和权值以外,还有一个访问频度。我们定义一个结点在树中的访问代价为它的访问频度乘以它在树中的深度。整棵树的访问代价定义为所有结点在树中的访问代价之和。
	现在给定每个结点的数据值、权值和访问频度,你可以根据需要修改某些结点的权值,但每次修改你会付出K的额外修改代价。你可以把结点的权值改为任何实数,但是修改后所有结点的权值必须仍保持互不相同。现在你要解决的问题是,整棵树的访问代价与额外修改代价的和最小是多少?
【输入文件】
输入文件名为treapmod.in。
输入文件第一行包含两个正整数N和K。N为结点的个数,K为每次修改所需的额外修改代价。
接下来一行包含N个非负整数,是每个结点的数据值。
再接下来一行包含N个非负整数,是每个结点的权值。
再接下来一行包含N个非负整数,是每个结点的访问频度。
所有的数据值、权值、访问频度均不超过400000。每两个数之间都有一个空格分隔,且行尾没有空格。
【输出文件】
输出文件名为treapmod.out。输出文件只有一个数字,即你所能得到的整棵树的访问代价与额外修改代价之和的最小值。
【输入样例】
4 10
1 2 3 4
1 2 3 4
1 2 3 4
【输出样例】
29
【样例说明】
输入的原图是左图,它的访问代价是1×1+2×2+3×3+4×4=30。最佳的修改方案是把输入中的第3个结点的权值改成0,得到右图,访问代价是1×2+2×3+3×1+4×2=19,加上额外修改代价10,一共是29。

【数据规模】
40%的数据满足N ≤ 30;
70%的数据满足N ≤ 50;
100%的数据满足N ≤ 70, 1 ≤ K ≤ 30000000。
这道题咋看似乎有点像Treap(但我不会Treap),所以想了很久Treap的各种操作。
后来发现,所有元素的左右相对位置并不会变,变化的只是一些结点的深度。并且,所有元素的访问代价和,等于分别以每个结点为根的所有树的权值和。
于是可以考虑区间DP。设weight[k]为结点k离散化过后的权值,那么转移方程如下:


Accode:

#include <cstdio>
#include <cstdlib>
#include <algorithm>
#include <cstring>
#include <string>

typedef long long int64;
const int maxN = 80;
const int64 INF = 0x3f3f3f3f3f3f3f3fLL;
struct Node {int val, wei, freq;} node[maxN];
int64 f[maxN][maxN][maxN], sum[maxN]; int n, K;

inline int getint()
{
	int res = 0; char tmp;
	while (!isdigit(tmp = getchar()));
	do res = (res << 3) + (res << 1) + tmp - '0';
	while (isdigit(tmp = getchar()));
	return res;
}

inline bool cmp1(const Node &a, const Node &b)
{return a.wei < b.wei;}

inline bool cmp2(const Node &a, const Node &b)
{return a.val < b.val;}

inline int64 s(int i, int j) {return sum[j] - sum[i - 1];}

int64 Dp(int i, int j, int w)
{
	if (i > j) return 0;
	if (f[i][j][w] + 1) return f[i][j][w];
	f[i][j][w] = INF;
	//注意这里必须是先赋值为正无穷,后逐渐松弛到最小。
	for (int k = i; k < j + 1; ++k)
	{
		f[i][j][w] = std::min(f[i][j][w], Dp(i, k - 1, w) +
			Dp(k + 1, j, w) + s(i, j) + K);
		if (node[k].wei >= w)
			f[i][j][w] = std::min(f[i][j][w],
				Dp(i, k - 1, node[k].wei) +
				Dp(k + 1, j, node[k].wei)
				+ s(i, j));
	}
	return f[i][j][w];
}

inline void printI64(int64 x)
{
	static const int64 MOD = 1000000000;
	if (x < MOD) printf("%d", (int)x);
	else printf("%d%09d", int(x / MOD), int(x % MOD));
	return;
}

int main()
{
	freopen("treapmod.in", "r", stdin);
	freopen("treapmod.out", "w", stdout);
	n = getint(); K = getint();
	for (int i = 1; i < n + 1; ++i) node[i].val = getint();
	for (int i = 1; i < n + 1; ++i) node[i].wei = getint();
	for (int i = 1; i < n + 1; ++i) node[i].freq = getint();
	std::sort(node + 1, node + n + 1, cmp1);
	for (int i = 1; i < n + 1; ++i) node[i].wei = i;
	std::sort(node + 1, node + n + 1, cmp2);
	for (int i = 1; i < n + 1; ++i) 
		sum[i] = sum[i - 1] + node[i].freq;
	memset(f, 0xff, sizeof f);
	printI64(Dp(1, n, 0)); printf("\n");
	return 0;
}

  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值