[luogu1486][鬱悶的出納員]

wxyww發表於2018-12-02

題目連結

思路

這個題其實就是對於treap中的刪除操作進行一些修改。自己yy了一種做法。就是在刪除的時候，如果要刪除的數比這棵子樹的根大，那麼就把根變成根的右孩子，這樣就相當於刪除了整棵左子樹和根節點。然後重新維護一下siz，並且維護一下平衡性就行了。
竟然把rotate函式寫錯了。調了30min。又沒正確理解

如果某個員工的初始工資低於最低工資標準,那麼將不計入最後的答案內
這句話的含義，又調了30min。23333

程式碼

//每次修改操作之後都進行一次刪除，並且更改刪除函式
/*
* @Author: wxyww
* @Date:   2018-12-02 08:41:38
* @Last Modified time: 2018-12-02 10:02:49
*/
#include<cstdio>
#include<iostream>
#include<cstdlib>
#include<cmath>
#include<ctime>
#include<bitset>
using namespace std;
#define ls TR[cur].ch[0]
#define rs TR[cur].ch[1]
typedef long long ll;
const int N = 100000 + 100,INF = 1e9 + 7;
ll read() {
   ll x=0,f=1;char c=getchar();
   while(c<`0`||c>`9`) {
      if(c==`-`) f=-1;
      c=getchar();
   }
   while(c>=`0`&&c<=`9`) {
      x=x*10+c-`0`;
      c=getchar();
   }
   return x*f;
}
struct node {
   int ch[2],id,val,siz,cnt;
}TR[N];
int MIN;
void up(int cur) {
   TR[cur].siz = TR[ls].siz + TR[rs].siz + TR[cur].cnt;
}
void rotate(int &cur,int f) {
   int son = TR[cur].ch[f];
   TR[cur].ch[f] = TR[son].ch[f ^ 1];
   TR[son].ch[f ^ 1] = cur;
   up(cur);
   cur = son;
   up(cur);
}
int tot;
void insert(int &cur,int val) {
   if(!cur) {
      cur = ++tot;
      TR[cur].id = rand();
      TR[cur].val = val;
      TR[cur].siz = TR[cur].cnt = 1;
      return;
   }
   TR[cur].siz++;
   if(val == TR[cur].val) {TR[cur].cnt++;return;}
   int d = val > TR[cur].val;
   insert(TR[cur].ch[d],val);
   if(TR[TR[cur].ch[d]].id < TR[cur].id) rotate(cur,d);
}
void del(int &cur,int val) {
   if(!cur) return;
   if(val <= TR[cur].val) del(ls,val);
   else {
      del(rs,val);
      cur = rs;
   }
   up(cur);
   if(TR[ls].id < TR[cur].id && ls) rotate(cur,0);
   if(TR[rs].id < TR[cur].id && rs) rotate(cur,1);
}
int kth(int cur,int now) {
   while(1) {
      if(now > TR[rs].siz + TR[cur].cnt) now -= TR[rs].siz + TR[cur].cnt,cur = ls;
      else if(now <= TR[rs].siz) cur = rs;
      else return TR[cur].val; 
   }
}
int main() {
   int rt = 0;
   int n = read(),change = 0,num = 0;
   MIN = read();
   char c;
   while(n--) {
      int x;
      scanf("
%c %d",&c,&x);
      if(c == `I`) {
         x -= change;
         if(x >= MIN) num++,insert(rt,x);
      }
      if(c == `A`) MIN -= x,change += x;
      if(c == `S`) {
         MIN += x;
         del(rt,MIN);
         change -= x;
      }
      if(c == `F`) {
         if(x > TR[rt].siz) puts("-1");
         else printf("%d
",kth(rt,x) + change);
      }
   }
   cout<<num - TR[rt].siz<<endl;
   return 0;
}

一言

時間會把你欠下的對不起，變成還不起，又會把很多對不起，變成來不及。 ——乖，摸摸頭

相關文章