[BZOJ1564][NOI2009]二叉查找树

[BZOJ1564][NOI2009]二叉查找树

试题描述

输入

输出

只有一个数字，即你所能得到的整棵树的访问代价与额外修改代价之和的最小值。

输入示例

4 10
1 2 3 4
1 2 3 4
1 2 3 4

输出示例

29

数据规模及约定

\(n\) 开成 \(110\) 足够了。

题解

先将节点按照中序遍历（即数据值升序）排好，把权值离散一下，然后可以设计 dp 了。令 \(f(l, r, k)\) 表示对于区间 \([l, r]\) 中的节点，权值都不小于 \(k\) 的最小访问代价。那么转移就是枚举哪个点作为根节点，然后看它是否需要改动权值，是要改大还是改小。

#include <iostream>
#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <cctype>
#include <algorithm>
using namespace std;
#define rep(i, s, t) for(int i = (s); i <= (t); i++)
#define dwn(i, s, t) for(int i = (s); i >= (t); i--)

int read() {
	int x = 0, f = 1; char c = getchar();
	while(!isdigit(c)){ if(c == '-') f = -1; c = getchar(); }
	while(isdigit(c)){ x = x * 10 + c - '0'; c = getchar(); }
	return x * f;
}

#define maxn 110
#define oo 2147483647

struct Node {
	int key, val, freq;
	Node() {}
	Node(int _1, int _2, int _3): key(_1), val(_2), freq(_3) {}
	bool operator < (const Node &t) const { return key < t.key; }
} ns[maxn];
int n, K, num[maxn], f[maxn][maxn][maxn];

int dp(int l, int r, int lo) {
	if(l == r) return ns[l].freq + K * (ns[l].val < lo);
	if(l > r) return 0;
	if(f[l][r][lo] < oo) return f[l][r][lo];
	int sum = 0;
	rep(i, l, r) sum += ns[i].freq;
	rep(i, l, r)
		f[l][r][lo] = min(f[l][r][lo], dp(l, i - 1, max(lo, ns[i].val + 1)) + dp(i + 1, r, max(lo, ns[i].val + 1)) + sum + K * (ns[i].val < lo)),
		f[l][r][lo] = min(f[l][r][lo], dp(l, i - 1, lo) + dp(i + 1, r, lo) + sum + K * (ns[i].val != lo));
	// printf("f[%d][%d][%d] = %d\n", l, r, lo, f[l][r][lo]);
	return f[l][r][lo];
}

int main() {
	n = read(); K = read();
	rep(i, 1, n) ns[i].key = read();
	rep(i, 1, n) num[i] = ns[i].val = read(); num[n+1] = oo;
	rep(i, 1, n) ns[i].freq = read();
	sort(ns + 1, ns + n + 1);
	
	rep(i, 1, n) rep(j, 1, n) rep(k, 1, n + 1) f[i][j][k] = oo;
	sort(num + 1, num + n + 1);
	rep(i, 1, n) ns[i].val = lower_bound(num + 1, num + n + 2, ns[i].val) - num;
	
	printf("%d\n", dp(1, n, 1));
	
	return 0;
}