码迷,mamicode.com
首页 > 其他好文 > 详细

【Bzoj1503】郁闷的出纳员(Splay)

时间:2018-01-31 22:26:44      阅读:339      评论:0      收藏:0      [点我收藏+]

标签:修改   排序   esc   统计   log   tips   相同   软件   clu   

Description

OIER公司是一家大型专业化软件公司,有着数以万计的员工。作为一名出纳员,我的任务之一便是统计每位员工的工资。这本来是一份不错的工作,但是令人郁闷的是,我们的老板反复无常,经常调整员工的工资。如果他心情好,就可能把每位员工的工资加上一个相同的量。反之,如果心情不好,就可能把他们的工资扣除一个相同的量。我真不知道除了调工资他还做什么其它事情。

工资的频繁调整很让员工反感,尤其是集体扣除工资的时候,一旦某位员工发现自己的工资已经低于了合同规定的工资下界,他就会立刻气愤地离开公司,并且再也不会回来了。每位员工的工资下界都是统一规定的。每当一个人离开公司,我就要从电脑中把他的工资档案删去,同样,每当公司招聘了一位新员工,我就得为他新建一个工资档案。

老板经常到我这边来询问工资情况,他并不问具体某位员工的工资情况,而是问现在工资第k多的员工拿多少工资。每当这时,我就不得不对数万个员工进行一次漫长的排序,然后告诉他答案。好了,现在你已经对我的工作了解不少了。正如你猜的那样,我想请你编一个工资统计程序。怎么样,不是很困难吧?

PS:最后还要求出踢出员工的数量

Solution

首先这题要求数列第K大的数,涉及到插入、删除和修改,

修改是对于整个数列,显然不能一个一个改,记录一个类似lazy_tag的标记即可

插入的时候将Num减去tag再插入

全体增加直接累加tag就行,但全体减少的话可能会踢人,进行删除操作判断

在删除的时候统计删除节点数目

Tips:插入对于s(u)的维护,需要考虑到父节点们,被坑了好久

Code

#include <cstdio>
#include <algorithm>
#define N 100010
#define lc(x) T[(x)][0]
using namespace std;

int n,m,rt,tot,T[N][2],k[N],fa[N],s[N],tag,num;

void rotate(int p){
    int q=fa[p],y=fa[q],x=(T[q][1]==p);
    T[q][x]=T[p][x^1],fa[T[q][x]]=q;
    T[p][x^1]=q,fa[q]=p;
    fa[p]=y;
    if(y){
        if(T[y][0]==q) T[y][0]=p;
        else if(T[y][1]==q) T[y][1]=p;
    }
    s[p]=s[q];
    s[q]=s[T[q][0]]+s[T[q][1]]+1;
}

void splay(int x){
    for(int y;y=fa[x];rotate(x))
        if(fa[y]) rotate((x==lc(y))==(y==lc(fa[y]))?y:x);
    rt=x;
}

void Insert(int v){
    if(!rt){
        rt=++tot;
        T[rt][0]=T[rt][1]=0;
        k[rt]=v;
        fa[rt]=0;
        s[rt]=1;
        return;
    }   
    int x=rt,y;
    while(1){
        s[x]++;
        y=T[x][k[x]<v];
        if(!y){
            y=++tot;
            k[y]=v;
            fa[y]=x;
            s[y]=1;
            T[y][0]=T[y][1]=0;
            T[x][k[x]<v]=y;
            break;
        }
        x=y;
    }
    splay(y);
}

void Del(int x){
    splay(x);
    num+=s[T[x][0]]+1;
    rt=T[x][1];
    fa[rt]=0;
}

void DEL(){
    int x=rt;
    while(x){
        int u=0;
        if(k[x]+tag>=m) x=T[x][0];
        else u=x,x=T[x][1];
        if(u) Del(u);
    }
}

int Find(int x){
    int r=0;
    for(int u=rt;;){
        if(s[T[u][1]]+r+1==x) return k[u];
        if(s[T[u][1]]+r+1<x) r+=s[T[u][1]]+1,u=T[u][0];
        else u=T[u][1];
    }
}

int main(){
    scanf("%d%d",&n,&m);
    while(n--){
        char ch=getchar();int t;
        while(ch!='I'&&ch!='A'&&ch!='S'&&ch!='F') ch=getchar();
        scanf("%d",&t);
        if(ch=='I'&&t>=m) Insert(t-tag);
        else if(ch=='A') tag+=t;
        else if(ch=='S') {tag-=t;DEL();}
        else if(ch=='F'){
            if(t>s[rt]) printf("-1\n");
            else printf("%d\n",Find(t)+tag);
        }
    }
    printf("%d\n",num);
    return 0;
}

【Bzoj1503】郁闷的出纳员(Splay)

标签:修改   排序   esc   统计   log   tips   相同   软件   clu   

原文地址:https://www.cnblogs.com/void-f/p/8394795.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!