poj3016 K-Monotonic，左偏树，dp

今天学习了一下左偏树（Leftist Tree）。

左偏树是个二叉树，满足堆性质，它最神奇的地方就是右边比较短(O(logn))。

嗯，然后他就可以往最右边递归合并，这样合并的复杂度也就是O(logn)级别

插入就是将要插入节点作为一个树合并。弹出呢，只要把根节点的两个儿子合并一下就好了。

 

貌似这个东西应用不是很多啊。

做了个poj3016练手。

这个题问的是将一个数列变成k个严格单调串需要的总变化值的最小值。

 

嗯左偏树最简单的应用就是将一个数列变成单调非减，求最小变化值的总和。

它的算法请搜黄源河2005国家队论文。

具体说来就是

对于每个i,1<=i<=n

1.将i作为一个区间，放入栈中

2.如果栈顶中位数小于次栈顶的中位数，那么将栈顶与次顶的两个区间合并，维护中位数。并继续此步骤。

突然发现这题好像多校6的1003题（hdu4923）。做法都很像。

不过这里要用左偏树合并两个区间，维护中位数。

 

至于本题，只需要从前往后再从后往前用上述做法做一遍，求出一个区间最小的cost[i][j]，cost[i][j]表示将i到j合并成单调的一段所需最小费用。

然后可以用一个简单的dp：f[i][k]=min{f[j-1][k-1]+cost[j][i]}来得到答案，f[i][j]表示将1..i都合并为单调，合并了j次的最小费用。

 

这题还有一个技巧，a[i-1]<a[i] -> a[i-1]+1<=a[i] -> a[i-1]-(i-1)<=a[i]-i->b[i-1]<=b[i]，(b[i]=a[i]-i)

这样就从严格单调转化为了非严格单调。

#include<iostream>
#include<cstdio>
#include<cstring>
#include<cstdlib>
using namespace std;
#define Maxn 1100

//--
struct LeftistTree{
    int l,r,dis,v,cnt;
    void init(int val){
        l=r=dis=0;v=val;cnt=1;
    }
}lt[Maxn];
int tt;

int merge(int a,int b){
    if (a==0) return b;
    if (b==0) return a;
    if (lt[a].v<lt[b].v) swap(a,b);
    lt[a].r=merge(lt[a].r,b);
    if (lt[lt[a].l].dis<lt[lt[a].r].dis) swap(lt[a].l,lt[a].r);
    if (lt[a].r==0) {lt[a].dis=0;}
    else {lt[a].dis=lt[lt[a].r].dis+1;}
    return a;
}

int gettop(int a){
    return lt[a].v;
}

int pop(int a){
    int l=lt[a].l,r=lt[a].r;
    lt[a].init(0);
    return merge(l,r);
}

int insert(int root,int nval){
    lt[++tt].init(nval);
    return merge(root,tt);
}

void initLT(){
    int i;
    tt=0;
    for(i=1;i<=Maxn-10;++i){
        lt[i].init(0);
    }
}
//--
int costb[Maxn][Maxn],costc[Maxn][Maxn];
int b[Maxn],c[Maxn];
int val[Maxn];
int sta[Maxn],pos[Maxn],smallsum[Maxn],smallcnt[Maxn],bigsum[Maxn],bigcnt[Maxn];
int n,t;

void prework(int v[],int cost[Maxn][Maxn]){
    int i,j,tv,top,tpos,tcnt;
    for(i=1;i<=n;++i){
        top=0;
        initLT();
        cost[i][i-1]=0;
        pos[0]=i-1;
        smallsum[0]=0;
        smallcnt[0]=0;
        bigsum[0]=0;
        bigcnt[0]=0;
        for(j=i;j<=n;++j){
            lt[++tt].init(v[j]);
            sta[++top]=tt;
            pos[top]=j;
            smallsum[top]=v[j];
            smallcnt[top]=1;
            bigsum[top]=0;
            bigcnt[top]=0;
            while(top>1){
                if (lt[sta[top]].v<lt[sta[top-1]].v){
                    sta[top]=merge(sta[top],sta[top-1]);
                    smallsum[top-1]+=smallsum[top];
                    smallcnt[top-1]+=smallcnt[top];
                    bigsum[top-1]+=bigsum[top];
                    bigcnt[top-1]+=bigcnt[top];
                    top--;
                    tcnt=(j+2-pos[top])/2;
                    while(smallcnt[top]>tcnt){
                        tv=gettop(sta[top]);
                        sta[top]=pop(sta[top]);
                        smallcnt[top]-=1;smallsum[top]-=tv;
                        bigcnt[top]+=1;bigsum[top]+=tv;
                    }
                }
                else break;
            }
            tpos=pos[top]-1;
            tv=gettop(sta[top]);
            cost[i][j]=cost[i][tpos]+smallcnt[top]*tv-smallsum[top]+bigsum[top]-tv*bigcnt[top];
        }

    }
}

int f[Maxn][Maxn];

int main(){
    //freopen("3016in.txt","r",stdin);
    int i,j,k;
    int inf;
    while(scanf("%d%d",&n,&t)&&(n||t)){
        for(i=1;i<=n;++i){
            scanf("%d",&val[i]);
            b[i]=val[i]-i;
            c[n+1-i]=val[i]-(n+1-i);
        }

        prework(b,costb);
        prework(c,costc);
        for(i=1;i<=n;++i){
            for(j=i;j<=n;++j){
                costb[i][j]=min(costb[i][j],costc[n+1-j][n+1-i]);
            }
        }
        inf=500000000;
        for(i=0;i<=n;++i)
            for(j=0;j<=n;++j) f[i][j]=inf;


        f[0][0]=0;
        for(i=1;i<=n;++i){
            for(j=1;j<=t;++j){
                for(k=1;k<=i;++k){
                    if (f[k-1][j-1]+costb[k][i]<f[i][j]) f[i][j]=f[k-1][j-1]+costb[k][i];
                }
            }
        }
        int ans=inf;
        for(i=1;i<=t;++i)if (ans>f[n][i]) ans=f[n][i];
        printf("%d\n",ans);
    }
    return 0;
}