poj 3666 Making the Grade

http://poj.org/problem?id=3666

这是一道 练习左偏树的好题 不过很多人都是用动态规划做的

没看过左偏树的建议看看 课件之类的 不难理解 有一份ppt 上有原题 

代码：

#include<iostream>

#include<cstdio>

#include<cstring>

#include<algorithm>

#include<string>

#include<vector>

#include<set>

#include<queue>

#include<stack>

#include<cmath>

#define LL long long



using namespace std;



const int N=2005;

struct node

{

    int k,d;

    struct node *l,*r;

};

struct node *head[N];//每个左偏树的根节点  大顶堆

int I;

int l[N],r[N];//每个左偏树的左右范围 不过我们这里只是保存了中位数(如果有n个元素 中位数我们这里定义为第 n/2个数) 和小于中位数的数

int a[N];

void  Left_tree_Merge(struct node *t1,struct node *t2)//将 t2指向的树合并到 t1

{

    struct node *t=t1->r;

    t1->r=t2;

    if(t==NULL)

    return;

    struct node *w=t1;

    while(w->r!=NULL&&w->r->k>t->k)//找到需要插入的地方 为 w的右孩子

    w=w->r;

    if(w->r==NULL)

    {w->r=t;}//为空直接插入

    else//否则递归合并

    {

        struct node *temp=w->r;

        w->r=t;

        Left_tree_Merge(t,temp);

    }

    return ;

}

struct node * Left_tree_pop(struct node *t)//弹出堆顶元素 

{

    if(t->l==NULL&&t->r!=NULL)

    {return t->r;}

    if(t->l!=NULL&&t->r==NULL)

    {return t->l;}

    if(t->l->k<t->r->k)

    swap(t->l,t->r);

    Left_tree_Merge(t->l,t->r);

    return t->l;

}

int update(struct node *t)//更新节点的 距离 维护左偏性

{

   if(t==NULL)

   return -1;

   int l1=update(t->l);

   int l2=update(t->r);

   t->d=min(l1,l2)+1;

   if(l1<l2)

   swap(t->l,t->r);

   return t->d;

}

int solve(int n)

{

    int I=0;

    for(int i=0;i<n;++i,++I)

    {

        head[I]=new node;

        head[I]->d=0;

        head[I]->l=head[I]->r=NULL;

        head[I]->k=a[i];

        l[I]=r[I]=i;

        while(I>0&&head[I-1]->k>=head[I]->k)//如果需要合并

        {

            Left_tree_Merge(head[I-1],head[I]);

            update(head[I-1]);//合并后更新

            if((r[I]-l[I]+1)%2==1&&(r[I-1]-l[I-1]+1)%2==1)//如果合并后元素过多 减少出一个 然后更新

            {

                head[I-1]=Left_tree_pop(head[I-1]);

                update(head[I-1]);

            }

            r[I-1]=r[I];//合并区间

            --I;

        }

    }

    int ans=0;

    for(int i=0;i<I;++i)

    {

        for(int j=l[i];j<=r[i];++j)

        {

            ans+=(abs(head[i]->k-a[j]));//累加

        }

    }

    return ans;

}

int main()

{

    //freopen("data.txt","r",stdin);

    int n;

    while(scanf("%d",&n)!=EOF)

    {

        for(int i=0;i<n;++i)

        scanf("%d",&a[i]);//先求非递减的解

        int ans=solve(n);

        for(int i=0;i<n/2;++i)

        {

            swap(a[i],a[n-i-1]);//将数组逆序后 再求一边便是 非递加的了

        }

        ans=min(ans,solve(n));

        printf("%d\n",ans);

    }

    return 0;

}