题目链接
http://www.lydsy.com/JudgeOnline/problem.php?id=1564
思路
看上去此题似乎并不好做,但是注意到,这个二叉查找树就是一个treap,而且这个treap在权值上看是个小根堆,在键值上看,键值的前序遍历是升序的。因此该二叉查找树的前序遍历显然是固定的。而且由于初始时每个结点的键值和权值互不相同,且要求修改后权值仍然互不相同,实际上我们可以认为修改后完全可以保证权值互不相同。
首先我们对所有点的权值进行离散化,这样每个点的权值必然在区间 [1,n] 中,用 f[i][j][w] 来表示前序遍历中,区间 [i,j] 对应的点作为一个子树,并且子树中所有权值都大于等于 w ,这个子树访问代价与修改代价之和的最小值。
显然我们要枚举这个子树的根节点
因此可以得到下面的DP方程:
DP时要保证在DP当前树时,先DP完这个树的所有子树。这样最终的答案里,深度为
d
的点的频率恰好给最终答案计算了
当然DP也有边界条件,
f[i][i−1][w]=0
,因为
f[i][i−1][w]
就是表示点
i
为根结点,下面无左儿子(或
另外注意枚举子树时,显然DP是要先求小子树的
f
值,再求包含了它的大树的
直接这样做复杂度是 O(n 5 ) ,但是求和那部分可以通过前缀和优化,因此可以把复杂度降低到 O(n 4 )
代码
#include <iostream>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <algorithm>
#define MAXN 80
using namespace std;
struct Node
{
int val; //键值
int weight; //权值
int frequence; //频率
}node[MAXN];
int n,K;
bool cmp(Node a,Node b)
{
return a.val<b.val;
}
int f[MAXN][MAXN][MAXN],stack[MAXN],top=0; //边界:f[i][i-1][w]=0,表示点i为根结点,下面无左儿子(i-1为根节点,下面无右儿子),显然f值为0
int sum[MAXN]; //频率的前缀和
int main()
{
scanf("%d%d",&n,&K);
for(int i=1;i<=n;i++)
scanf("%d",&node[i].val);
for(int i=1;i<=n;i++)
scanf("%d",&node[i].weight),stack[++top]=node[i].weight;
for(int i=1;i<=n;i++)
scanf("%d",&node[i].frequence);
sort(stack+1,stack+top+1);
for(int i=1;i<=n;i++)
node[i].weight=lower_bound(stack+1,stack+top+1,node[i].weight)-stack;
sort(node+1,node+n+1,cmp);
for(int i=1;i<=n;i++)
sum[i]=sum[i-1]+node[i].frequence;
memset(f,0x3f,sizeof(f));
for(int i=1;i<=n+1;i++)
for(int w=0;w<=n;w++)
f[i][i-1][w]=0;
for(int w=n;w>=1;w--)
for(int i=n;i>=1;i--)
for(int j=i;j<=n;j++)
for(int k=i;k<=j;k++) //枚举[i,j]这段区间的点对应的子树,该子树以点k作为根节点
{
f[i][j][w]=min(f[i][j][w],f[i][k-1][w]+f[k+1][j][w]+K+sum[j]-sum[i-1]); //k本来权值小于K,将其权值变为k
if(node[k].weight>=w) f[i][j][w]=min(f[i][j][w],f[i][k-1][node[k].weight]+f[k+1][j][node[k].weight]+sum[j]-sum[i-1]); //k的权值本身就大于等于K,因此可以不变
}
int ans=0x3f3f3f3f;
for(int w=0;w<=n;w++)
ans=min(ans,f[1][n][w]);
printf("%d\n",ans);
return 0;
}