「SCOI2006」zh_tree

「SCOI2006」zh_tree

题目限制

• 内存限制：250.00MB
• 时间限制：1.00s
• 标准输入输出

题目知识点

• 思维
• 动态规划 $dp$
  • 区间$dp$

题目来源

「 洛谷 」P4539 [SCOI2006]zh_tree

题目

题目背景

张老师根据自己工作的需要，设计了一种特殊的二叉搜索树

题目描述

他把这种二叉树起名为 zh_tree，对于具有 $n$ 个节点的 zh_tree，其中序遍历恰好为 $(1、2、3、\cdot \cdot \cdot 、n)$，其中数字 $1、2、3、\cdot \cdot \cdot 、n$ 是每个节点的编号
$n$ 个结点恰好对应于一组学术论文中出现的 $n$ 个不同的单词

第 $i$ 个单词在该组论文中出现的次数记为 $d_i$，例如，$d_2=10$ 表示第 $2$ 个节点所对应的单词在该组论文中出现了 $10$ 次
设该组论文中出现的单词总数为 $S$，即 $S=d_1+d_2+\cdot \cdot \cdot +d_n={\sum }_{i=1}^n{d}_i$
记 $f_i=\frac{d_i}{S}$ 为第 $i$ 个单词在该组论文中出现的 概率(频率)

张老师把根节点深度规定为 $0$，如果第 $i$ 个节点的深度为 $r_i$，则访问该节点的代价为 $h_i$，$h_i=k(r_i+1)+c$，其中 $k、c$ 为已知的常数 $(0<k,c\le 100)$

则 zh_tree 是满足 $h_1{f}_1+h_2{f}_2+\cdot \cdot \cdot +h_n{f}_n={\sum }_{i=1}^n{h}_i{f}_i$ 最小的一棵二叉树
我们称上式为访问 zh_tree 的代价
请你根据已知的数据为张老师设计一棵 zh_tree

格式

输入格式

输入共 $2$ 行：
对于第 $1$ 行，$3$ 个用空格隔开的正数，$n、k、c$。其中 $n<30$，为整数；$k、c$ 为不超过 $100$ 的正实数
对于第 $2$ 行：$n$ 个用空格隔开的正整数，为每个单词出现的次数 $d_i(d_i<200)$

输出格式

输出一行：一个正实数，保留 $3$ 位小数，表示访问 zh_tree 的最小代价

样例

样例输入

4 2 3.5
20 30 50 20

样例输出

7.000

提示

数据范围

对于 $100\%$ 的数据：$1\le n<30$，$0<k、c\le 100$，$d_i<200$

---

思路

对于每个节点的深度 $r_i$，由于 $h_i=k(r_i+1)+c$，不妨可以把根节点的深度改为 $1$，这样就可以直接用 现在的 $r_i$ 表示 原来的 $r_i+1$ 了。

先把 $\sum _{i=1}^n{h}_i{f}_i$ 化简：
$$\begin{array}{rl}\sum _{i=1}^n(h_i\times f_i)& =\sum _{i=1}^n(k\times r_i+c)\times \frac{d_i}{S}\\ & =\frac{1}{S}\times \sum _{i=1}^n(k\times r_i+c)\times d_i\\ & =\frac{1}{S}\times \sum _{i=1}^n(k\times r_i\times d_i+c\times d_i)\\ & =\frac{1}{S}\times (\sum _{i=1}^{n}k\times r_i\times d_i+\sum _{i=1}^{n}c\times d_i)\\ & =(\frac{1}{S}\times \sum _{i=1}^{n}k\times r_i\times d_i)+(\frac{1}{S}\times c\times \sum _{i=1}^n{d}_i)\\ & =(\frac{k}{S}\times \sum _{i=1}^n{r}_i\times d_i)+(\frac{1}{S}\times c\times S)\\ & =\frac{k\times \sum _{i=1}^n{r}_i\times d_i}{S}+c\end{array}$$

题目要我们求 $\sum _{i=1}^n{h}_i{f}_i$ 的最小值，就可以转化成求 $\sum _{i=1}^n{r}_i\times d_i$ 的最小值。

• 状态定义：设 $f_{L,R}$ 表示：一棵二叉树的 中序遍历(左子树，根节点，右子树) 为 $L\sim R(L,L+1,L+2,\cdots ,R)$ 时 $\sum _{i=L}^R{r}_i\times d_i$ 的最小值，最终的答案就是 $\frac{k\times f_{1,n}}{S}+c$。
• 初始化：$\forall L=R=i,f_{L,R}=d_i$；$\forall L<R,f_{L,R}=+\infty$。
• 状态转移：对于中序遍历为 $L\sim R$ 的二叉树，枚举其根节点 $tr(L\le tr\le R)$， f L , R = min ⁡ t r = L R { f L , t r − 1 + f t r + 1 , R + ∑ i = L R d i } = min ⁡ t r = L R { f L , t r − 1 + f t r + 1 , R } + ∑ i = L R d i f_{L, R} = \min\limits_{tr = L}^R \{ f_{L, tr - 1} + f_{tr + 1, R} + \sum\limits_{i = L}^{R} d_i \} = \min\limits_{tr = L}^R \{ f_{L, tr - 1} + f_{tr + 1, R} \} + \sum\limits_{i = L}^{R} d_i fL,R​=tr=LminR​{fL,tr−1​+ftr+1,R​+i=L∑R​di​}=tr=LminR​{fL,tr−1​+ftr+1,R​}+i=L∑R​di​。

如果需要输出方案，可以在转移过程中记录下 $f_{L,R}$ 取最小值时的根节点，最后递归即可。

转移时 $\sum _{i=L}^R{d}_i$ 可以用前缀和作差，这样时间复杂度 $O(n^4)\to O(n^3)$，不过都可以 $\text{AC}$。

---

代码

#include <cstdio>
#include <cstring>

#define Min(a, b) ((a) < (b) ? (a) : (b))

const int MAXN = 30;

int N, S;
int D[MAXN + 5];
int Pre[MAXN + 5]; // 前缀和优化 

double K, C;
double dp[MAXN + 5][MAXN + 5];

int main()
{
	memset(dp, 0x3f, sizeof(dp));
	scanf("%d %lf %lf", &N, &K, &C);
	for (int i = 1; i <= N; i++)
	{
		scanf("%d", &D[i]);
		Pre[i] = S += D[i];
		dp[i][i] = 1.0 * D[i];
	}
	for (int len = 2; len <= N; len++)
	{
		for (int i = 1, j = len; j <= N; i++, j++)
		{
			dp[i][j] = Min(dp[i][j - 1], dp[i + 1][j]);
			for (int k = i + 1; k <= j - 1; k++)
				dp[i][j] = Min(dp[i][j], dp[i][k - 1] + dp[k + 1][j]);
			dp[i][j] += 1.0 * (Pre[j] - Pre[i - 1]);
		}
	}
	printf("%.3lf\n", K * dp[1][N] / S + C);
	return 0;
}

---