[BZOJ 1564][NOI 2009]二叉查找树(DP)

题目链接

http://www.lydsy.com/JudgeOnline/problem.php?id=1564

思路

看上去此题似乎并不好做，但是注意到，这个二叉查找树就是一个treap,而且这个treap在权值上看是个小根堆，在键值上看，键值的前序遍历是升序的。因此该二叉查找树的前序遍历显然是固定的。而且由于初始时每个结点的键值和权值互不相同，且要求修改后权值仍然互不相同，实际上我们可以认为修改后完全可以保证权值互不相同。

首先我们对所有点的权值进行离散化，这样每个点的权值必然在区间$[1,n]$中，用$f[i][j][w]$来表示前序遍历中，区间$[i,j]$对应的点作为一个子树，并且子树中所有权值都大于等于$w$，这个子树访问代价与修改代价之和的最小值。

显然我们要枚举这个子树的根节点$k$，那么有两种情况：1、点$k$本来的权值是小于$w$的，那么要符合小根堆的特性，点$k$权值要修改为$w$。2、点$k$本来的权值就是大于等于$w$的，那么可以选择不修改点$k$的权值，也可以选择修改点$k$的权值。

因此可以得到下面的DP方程：

$$f[i][j][w]=\min_{i\leq k \leq j} \{ f[i][k-1][w]+f[k+1][j][w]+K+\sum_{t=i}^jfrequence_t\},weight_k<w$$
$$f[i][j][w]=\min_{i\leq k \leq j} \{ f[i][k-1][weight_k]+f[k+1][j][weight_k]+\sum_{t=i}^jfrequence_t\}$$

DP时要保证在DP当前树时，先DP完这个树的所有子树。这样最终的答案里，深度为$d$的点的频率恰好给最终答案计算了$d$次贡献，满足了题目的要求。

当然DP也有边界条件，$f[i][i-1][w]=0$，因为$f[i][i-1][w]$就是表示点$i$为根结点，下面无左儿子(或$i-1$为根节点，下面无右儿子)的代价和最小值，显然f值为0

另外注意枚举子树时，显然DP是要先求小子树的$f$值，再求包含了它的大树的$f$值，因此要先在前序遍历序列里从右往左枚举$i$，再从$i到n$枚举$j$，最后从$i$到$j$枚举$k$

直接这样做复杂度是$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;
}