bzoj1503 郁闷的出纳员(平衡树,思维)

题目大意:
现在有n个操作和一个最低限度m

I 命令 I   k 新建一个工资档案,初始工资为k。
A 命令 A   k 把每位员工的工资加上k

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

F 命令 F   k 查询第k多的工资 (如果当前的员工总数不够k,就输出 1

其中 n 100000

最后还要输出最终离开了多少个员工

需要注意的是 ,如果某员工的初始工资低于工资下界,他将立刻离开公司!!!!

一看这个题,就感觉是个 s p l a y

不过emmmm 这个整体加和整体减应该怎么弄呢

我们考虑维护一个整体的变化值 t m p

那么对于splay中的每个元素,它的实际权值就是 x + t m p

这么来说

对于 I 操作,如果它的权值 > m,则 i n s e r t ( x t m p )

对于 A 操作,直接 t m p + = x

对于 S 操作,我们让 t m p = x ,并且判断有没有需要退出的点(就是直接将 i n f 转到 r o o t ,然后把 m i n 1 t m p 1 的后继转到root的右儿子,删掉左儿子)这样做的原因是避免哨兵被删掉

对于 F 操作,那就直接返回第k大,然后 + t m p 就可以

代码还是有很多细节的,比如要维护当前平衡树中有几个元素,事先加入哨兵之类的,直接看代码吧

#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cstring>
#include<cmath>
#include<map>
#include<queue>
#include<vector>

using namespace std;

inline int read()
{
   int x=0,f=1;char ch=getchar();
   while (!isdigit(ch)){if (ch=='-') f=-1;ch=getchar();}
   while (isdigit(ch)){x=(x<<1)+(x<<3)+ch-'0';ch=getchar();}
   return x*f;
}

const int maxn = 1e6+1e2;

int ch[maxn][4];
int val[maxn],sz[maxn],size;
int fa[maxn];
int cnt[maxn];
int n,m,ymh,root,min1;
int tot;
int num=0;
int ans;

int son(int x)
{
    if (x==ch[fa[x]][0]) return 0;
    else return 1;
}
void update(int x)
{
    sz[x]=sz[ch[x][0]]+sz[ch[x][1]]+cnt[x];
}

void rotate(int x)
{
    int y=fa[x],z=fa[y];
    int b=son(x),c=son(y);
    ch[z][c]=x;
    fa[x]=z;
    ch[y][b]=ch[x][!b];
    fa[ch[x][!b]]=y;
    ch[x][!b]=y;
    fa[y]=x;
    update(y);
    update(x);
}

void splay(int x,int p)
{
    while (fa[x]!=p)
    {
        int y=fa[x],z=fa[y];
        if (z==p) rotate(x);
        else
        {
            if (son(x)==son(y))
            {
                rotate(y);
                rotate(x);
            }
            else
            {
                rotate(x);
                rotate(x);
            }
        }
    }
    if (fa[x]==0) root=x;
}

int find_qq(int x)
{
    int now = root,num=0;
    while (now)
    {
        if (val[now]<x)
        {
            num=now;
            now=ch[now][1];
        }
        else now=ch[now][0];
    }
    return num;
}

int find_hj(int x)
{
    int now = root,num=0;
    while (now)
    {
        if (val[now]>x)
        {
            num=now;
            now=ch[now][0];
        }
        else now=ch[now][1];
    }
    return num;
}

void insert(int x)
{
    tot++;
    int qq = find_qq(x);
    int hj = find_hj(x);
    splay(qq,0);
    splay(hj,qq);
    int y = ch[hj][0];
    if (cnt[y])
    {
        cnt[y]++;
        update(y);
    }
    else
    {
        ++size;
        ch[hj][0]=size;
        fa[size]=hj;
        val[size]=x;
        cnt[size]=1;
        sz[size]=1;
        //cout<<"gg11"<<endl;
        update(size);
    }
}

void delet(int x)
{
    --tot;
    int qq = find_qq(x);
    int hj = find_hj(x);
    splay(qq,0);
    splay(hj,qq);
    int y = ch[hj][0];
    if (cnt[y]>1)
    {
        cnt[y]--;
        update(y);
    }
    else
    {
        cnt[y]=0;
        fa[y]=0;
        ch[hj][0]=0;
        sz[y]=0;
        val[y]=0;   
    }
}

int kth(int x)
{
    int now = root;
    while (1)
    {
        if (x<=sz[ch[now][0]]) now=ch[now][0];
        else{
            if (x<=sz[ch[now][0]]+cnt[now]) return val[now];
            x-=sz[ch[now][0]]+cnt[now];
            now=ch[now][1]; 
          }
    }
}

int add;

int main()
{
  size=2;
  val[1]=2e9;
  val[2]=-2e9;
  fa[2]=1;
  ch[1][0]=2;
  root=1;
  scanf("%d%d",&n,&min1);
  for (int i=1;i<=n;i++)
  {
    char s[10];
    int x;
    scanf("%s",s+1);
    //cout<<size<<endl;
    x=read();
    //cout<<x<<endl;
    if (s[1]=='A')
    {
        ymh+=x;
      }
    if (s[1]=='S')
    {
        ymh-=x;
        int hj = find_hj(min1-ymh-1);
        splay(2,0);
        splay(hj,2);
        int pos = ch[hj][0];
        ans+=sz[pos];
        tot-=sz[pos];
        fa[pos]=0;
        ch[hj][0]=0;
        sz[pos]=0;
        cnt[pos]=0;
        /*if (tot==0) 
        {
        size=2;
            val[1]=2e9;
            val[2]=-2e9;
            fa[2]=1;
            ch[1][0]=2;
            root=1;
        }*/
        //cout<<ans;
    }
    if (s[1]=='I')
    {
        if (x<min1) continue;
        //cout<<"gg"<<endl;
        insert(x-ymh);
    }
    if (s[1]=='F')
    {
        //cout<<tot<<endl;
        if (x>tot)
          printf("-1\n");
        else
          printf("%d\n",kth(tot-x+1)+ymh); 
    }
    //cout<<tot<<endl;
  }
  cout<<ans;
  return 0;
}

猜你喜欢

转载自blog.csdn.net/y752742355/article/details/80566122