洛谷P1486 [NOI2004]鬱悶的出納員 平衡樹Splay
阿新 • • 發佈:2020-08-16
P1486 [NOI2004]鬱悶的出納員
題意
第一行有兩個整數 \(n\) 和 \(\min\)。\(n\) 表示下面有多少條命令,\(\min\) 表示工資下界。
接下來的 \(n\) 行,每行一個字元 \(x\) 和一個整數 \(k\),表示一條命令。命令可以是以下四種之一:
I k
新建一個工資檔案,初始工資為 \(k\)。如果某員工的初始工資低於工資下界,他將立刻離開公司。A k
把每位員工的工資加上 \(k\) 。S k
把每位員工的工資扣除 \(k\)。F k
查詢第 \(k\) 多的工資。
在初始時,可以認為公司裡一個員工也沒有。
分析
用\(delta\)表示所有員工的修改量,每個員工實際的工資為\(k+delta\)
對於新加進來的員工,為了消除歷史修改的影響,插入\(k-delta\),
對於增加工資操作,直接\(delta+=k\),
對於扣除工資操作,先\(delta-=k\),然後我們要找到所有\(k+delta<min\)的員工,移項一下即為\(k<min-delta\),可以先插入一個權值\(min-delta\),將其旋轉到根節點,左子樹即為所有要刪除的員工,直接將左子樹刪除即可,然後將\(min-delta\)刪除。
查詢第\(k\)多就是模板操作了。
Code
#include<algorithm> #include<iostream> #include<cstring> #include<iomanip> #include<sstream> #include<cstdio> #include<string> #include<vector> #include<bitset> #include<queue> #include<cmath> #include<stack> #include<set> #include<map> #define rep(i,x,n) for(int i=x;i<=n;i++) #define per(i,n,x) for(int i=n;i>=x;i--) #define sz(a) int(a.size()) #define rson mid+1,r,p<<1|1 #define pii pair<int,int> #define lson l,mid,p<<1 #define ll long long #define pb push_back #define mp make_pair #define se second #define fi first using namespace std; const double eps=1e-8; const int mod=1e9+7; const int N=3e5+10; const int inf=1e9; int n,m; int rt,tot,ch[N][2],sz[N],fa[N],cnt[N]; ll val[N],d; struct Splay{ void maintain(int x){ sz[x]=sz[ch[x][0]]+sz[ch[x][1]]+cnt[x]; } bool get(int x){ return x==ch[fa[x]][1]; } void clear(int x){ ch[x][0]=ch[x][1]=fa[x]=val[x]=sz[x]=cnt[x]=0; } void rotate(int x){ int y=fa[x],z=fa[y],chk=get(x); ch[y][chk]=ch[x][chk^1]; fa[ch[x][chk^1]]=y; ch[x][chk^1]=y; fa[y]=x; fa[x]=z; if(z) ch[z][ y==ch[z][1] ]=x; maintain(y); maintain(x); } void splay(int x,int goal=0){ for(int f;(f=fa[x])!=goal;rotate(x)){ if(fa[f]!=goal) rotate(get(x)==get(f)?f:x); } if(goal==0) rt=x; } void ins(ll k){ if(!rt){ val[++tot]=k; cnt[tot]++; rt=tot; maintain(rt); return; } int cnr=rt,f=0; while(1){ if(val[cnr]==k){ cnt[cnr]++; maintain(cnr); maintain(f); splay(cnr); break; } f=cnr; cnr=ch[cnr][val[cnr]<k]; if(!cnr){ val[++tot]=k; cnt[tot]++; fa[tot]=f; ch[f][val[f]<k]=tot; maintain(tot); maintain(f); splay(tot); break; } } } int rk(ll k){ int res=0,cnr=rt; while(1){ if(k<val[cnr]){ cnr=ch[cnr][0]; }else{ res+=sz[ch[cnr][0]]; if(k==val[cnr]){ splay(cnr); return res+1; } res+=cnt[cnr]; cnr=ch[cnr][1]; } } } ll kth(int k){ int cnr=rt; while(1){ if(ch[cnr][1]&&sz[ch[cnr][1]]>=k){ cnr=ch[cnr][1]; }else{ k-=cnt[cnr]+sz[ch[cnr][1]]; if(k<=0){ return val[cnr]; } cnr=ch[cnr][0]; } } } int pre(){ int cnr=ch[rt][0]; while(ch[cnr][1]) cnr=ch[cnr][1]; splay(cnr); return cnr; } void del(int k){ rk(k); if(cnt[rt]>1){ cnt[rt]--; maintain(rt); return; } if(!ch[rt][0]&&!ch[rt][1]){ clear(rt); rt=0; return; } if(!ch[rt][0]){ int cnr=rt; rt=ch[rt][1]; fa[rt]=0; clear(cnr); return; } if(!ch[rt][1]){ int cnr=rt; rt=ch[rt][0]; fa[rt]=0; clear(cnr); return; } int cnr=rt,x=pre(); splay(x); fa[ch[cnr][1]]=x; ch[x][1]=ch[cnr][1]; clear(cnr); maintain(rt); } }S; int main(){ //ios::sync_with_stdio(false); //freopen("in","r",stdin); scanf("%d%d",&n,&m); int cnt=0; rep(i,1,n){ char c[2]; int k; scanf("%s%d",c,&k); if(c[0]=='I'){ if(k>=m){ S.ins(k-d); cnt++; } }else if(c[0]=='A'){ d+=k; }else if(c[0]=='S'){ d-=k; S.ins(m-d); int pos=ch[rt][0]; S.clear(pos); ch[rt][0]=0; S.maintain(rt); S.del(m-d); }else{ if(sz[rt]<k) puts("-1"); else printf("%lld\n",S.kth(k)+d); } } printf("%d\n",cnt-sz[rt]); return 0; }