[BZOJ1564][NOI2009]二叉查找树 树形dp 区间dp

1564: [NOI2009]二叉查找树

Time Limit: 10 Sec  Memory Limit: 64 MB
Submit: 879  Solved: 612
[Submit][Status][Discuss]

Description

Input

Output

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

Sample Input

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

Sample Output

29

HINT

输入的原图是左图，它的访问代价是1×1+2×2+3×3+4×4=30。最佳的修改方案是把输入中的第3个结点的权值改成0，得到右图，访问代价是1×2+2×3+3×1+4×2=19，加上额外修改代价10，一共是29。

Source

观察到这就是一个treap。（其实没什么卵用）

设sum为访问频率的前缀和。

考虑dp，设f[w][i][j]表示从i到j组成根节点为原权值第w小之后的点的最小总代价。

第一位从原来权值最大的点开始枚举w。

之后两维枚举i，j。

下一维枚举分割点d。

分两种情况讨论：

1.若修改点d的权值，那么f[w][i][j]=min(f[w][i][j],f[w][i][d-1]+f[w][d+1][to]+sum[to]-sum[i-1]+k);

2.若不修改点d的权值，那么点d的权值要大于m，f[w][i][j]=min(f[w][i][j],f[t[d].a][i][d-1]+f[t[d].a][d+1][to]+sum[to]-sum[i-1]);

ans=f[1][1][n]

 #include<iostream>
 #include<cstring>
 #include<cstdio>
 #include<cstdlib>
 #include<cmath>
 #include<algorithm>
 #define maxn 80
 using namespace std;
 int read() {
     int x=,f=;char ch=getchar();
     for(;!isdigit(ch);ch=getchar()) if(ch=='-') f=-;
     for(;isdigit(ch);ch=getchar()) x=x*+ch-'';
     return x*f;
 }
 int n,k;
 struct data {
     int v,a,p;
 }t[maxn];
 bool cmp1(data t1,data t2) {return t1.a<t2.a;}
 bool cmp2(data t1,data t2) {return t1.v<t2.v;}
 int sum[maxn];
 int f[maxn][maxn][maxn];
 int main() {
     n=read();k=read();
     for(int i=;i<=n;i++) t[i].v=read();
     for(int i=;i<=n;i++) t[i].a=read();
     for(int i=;i<=n;i++) t[i].p=read();
     sort(t+,t+n+,cmp1);
     for(int i=;i<=n;i++) t[i].a=i;
     sort(t+,t+n+,cmp2);
     for(int i=;i<=n;i++) sum[i]=sum[i-]+t[i].p;
     for(int i=;i<=n;i++)
         for(int j=;j<=n;j++)
             if(t[i].a>=j) f[j][i][i]=t[i].p;
             else f[j][i][i]=t[i].p+k;
     for(int w=n;w>=;w--) {
         for(int j=;j<=n;j++) {
             for(int i=;i+j<=n;i++) {
                 int to=i+j;
                 int tmp=;
                 for(int d=i;d<=to;d++) {
                     if(t[d].a>=w) tmp=min(tmp,f[t[d].a][i][d-]+f[t[d].a][d+][to]+sum[to]-sum[i-]);
                     tmp=min(tmp,f[w][i][d-]+f[w][d+][to]+sum[to]-sum[i-]+k);
                 }
                 f[w][i][to]=tmp;
             }
         }
     }
     printf("%d",f[][][n]);
 }