zoukankan      html  css  js  c++  java
  • BZOJ1503[NOI2004]郁闷的出纳员——treap

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

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

    老板经常到我这边来询问工资情况,他并不问具体某位员工的工资情况,而是问现在工资第k多的员工拿多少工资。每当这时,我就不得不对数万个员工进行一次漫长的排序,然后告诉他答案。

    好了,现在你已经对我的工作了解不少了。正如你猜的那样,我想请你编一个工资统计程序。怎么样,不是很困难吧?

    如果某个员工的初始工资低于最低工资标准,那么将不计入最后的答案内

    输入格式:

    第一行有两个非负整数n和min。n表示下面有多少条命令,min表示工资下界。

    接下来的n行,每行表示一条命令。命令可以是以下四种之一:

    名称 格式 作用

    I命令 I_k 新建一个工资档案,初始工资为k。如果某员工的初始工资低于工资下界,他将立刻离开公司。

    A命令 A_k 把每位员工的工资加上k

    S命令 S_k 把每位员工的工资扣除k

    F命令 F_k 查询第k多的工资

    _(下划线)表示一个空格,I命令、A命令、S命令中的k是一个非负整数,F命令中的k是一个正整数。

    在初始时,可以认为公司里一个员工也没有。

    输出格式:

    输出文件的行数为F命令的条数加一。

    对于每条F命令,你的程序要输出一行,仅包含一个整数,为当前工资第k多的员工所拿的工资数,如果k大于目前员工的数目,则输出-1。

    输出文件的最后一行包含一个整数,为离开公司的员工的总数。

    输入样例#1: 
    9 10
    I 60
    I 70
    S 50
    F 2
    I 30
    S 15
    A 5
    F 1
    F 2
    
    输出样例#1: 
    10
    20
    -1
    2
    

    说明

    I命令的条数不超过100000

    A命令和S命令的总条数不超过100

    F命令的条数不超过100000

    每次工资调整的调整量不超过1000

    新员工的工资不超过100000

    这道题是一道经典的平衡树练习题,题中一共有四种操作:插入,查询,修改(上调和下降)。

    这里面唯一和一般的treap不一样的操作就是修改,如果每一次都把每个数加上或减掉一个值显然是不行的,但我们发现每次修改都是对全体修改,那么我们就可以不进行修改,相对的把工资下界调整,这样也就相当于修改了每个员工的工资。

    但是如果在某次涨工资之后来了一名员工,他并没有赶上涨工资,如果他的工资大于初始工资下界(这样才能进来)但却小于当前工资下界,那么下一次降工资后他本来不会跳槽(下降之后仍大于初始工资下界),却因为小于当前工资下界,那他到底会不会跳槽呢?依据题意是不会的,但如果按上述做法就是会跳槽的。那么就要在他刚进公司时我们要先给他加上之前涨的工资(使他工资一定大于当前工资下界),在查询到他时再减掉就可以了。

    工资下降相当于工资下界上调,这样就有可能有人会跳槽,因为I命令不超过10^5,树中最多不会超过10^5个节点,删除点个数也不会超过这些,所以暴力删除就好。

    如果查前驱函数是void函数要记得每次查询之后把answer归零。

    #include<cstdio>
    #include<algorithm>
    #include<iostream>
    #include<cmath>
    #include<ctime>
    #include<cstring>
    using namespace std;
    int r[100010];
    int v[100010];
    int w[100010];
    int ls[100010];
    int rs[100010];
    int size[100010];
    int tot;
    int root;
    int n,m;
    int del;
    int mini;
    int x;
    char a[10];
    int answer;
    int ans;
    void up(int x)
    {
        size[x]=size[rs[x]]+size[ls[x]]+w[x];
    }
    void rturn(int &x)
    {
        int t;
        t=ls[x];
        ls[x]=rs[t];
        rs[t]=x;
        size[t]=size[x];
        up(x);
        x=t;
    }
    void lturn(int &x)
    {
        int t;
        t=rs[x];
        rs[x]=ls[t];
        ls[t]=x;
        size[t]=size[x];
        up(x);
        x=t;
    }
    void insert_sum(int x,int &i)
    {
        if(!i)
        {
            i=++tot;
            w[i]=size[i]=1;
            v[i]=x;
            r[i]=rand();
            return ;
        }
        size[i]++;
        if(x==v[i])
        {
            w[i]++;
        }
        else if(x>v[i])
        {
            insert_sum(x,rs[i]);
            if(r[rs[i]]<r[i])
            {
                lturn(i);
            }
        }
        else
        {
            insert_sum(x,ls[i]);
            if(r[ls[i]]<r[i])    
            {
                rturn(i);
            }
        }
          
        return ;
    }
    void delete_sum(int x,int &i)
    {
        if(i==0)
        {
            return ;
        }
        if(v[i]==x)
        {
            if(w[i]>1)
            {
                w[i]--;
                size[i]--;
                return ;
            }
            if((ls[i]*rs[i])==0)
            {
                i=ls[i]+rs[i];
            }
            else if(r[ls[i]]<r[rs[i]])
            {
                rturn(i);
                delete_sum(x,i);
            }
            else
            {
                lturn(i);
                delete_sum(x,i);
            }
            return ;
        }
        size[i]--;
        if(v[i]<x)
        {
            delete_sum(x,rs[i]);
        }
        else
        {
            delete_sum(x,ls[i]);
        }
        return ;
    }
    int ask_sum(int x,int i)
    {
        if(i==0)
        {
            return 0;
        }
        if(x>size[ls[i]]+w[i])
        {
            return ask_sum(x-size[ls[i]]-w[i],rs[i]);
        }
        else if(size[ls[i]]>=x)
        {
            return ask_sum(x,ls[i]);
        }
        else
        {
            return v[i];
        }
    }
    void ask_front(int x,int i)
    {
        if(i==0)
        {
            return ;
        }
        if(v[i]<x)
        {
            answer=i;
            ask_front(x,rs[i]);
            return ;
        }
        else
        {
            ask_front(x,ls[i]);
            return ;
        }
        return ;
    }
    int main()
    {
        srand(12378);
        scanf("%d%d",&n,&m);
        mini=m;
        for(int i=1;i<=n;i++)
        {
            scanf("%s",a+1);
            scanf("%d",&x);
            if(a[1]=='I')
            {
                if(x>=mini)
                {
                    insert_sum(x-del,root);
                }
            }
            else if(a[1]=='A')
            {
                m-=x;
                del+=x;
            }
            else if(a[1]=='S')
            {
                m+=x;
                del-=x;
                answer=0;
                ask_front(m,root);
                while(answer!=0)
                {
                    delete_sum(v[answer],root);
                    ans++;
                    answer=0;
                    ask_front(m,root);
                     
                }
            }
            else if(a[1]=='F')
            {
                if(size[root]-x+1>0)
                {
                    printf("%d
    ",ask_sum(size[root]-x+1,root)+del);
                }
                else
                {
                    printf("-1
    ");
                }
            }
        }
        printf("%d",ans);
    }
  • 相关阅读:
    C++ 练习02 魔术师发牌问题
    C++ 入门2 类型转换
    C++ 入门1 C++简介
    一个简单计算器的实现
    C++练习01 打印杨辉三角
    数据结构01数据结构基础01
    Lesson_7 作业_1 Driver 和 Car
    Caterl Java寒假基础练习题(一) 循环相加
    Lesson_9 上课笔记 多态
    Lesson_10 作业计算工资
  • 原文地址:https://www.cnblogs.com/Khada-Jhin/p/8987211.html
Copyright © 2011-2022 走看看