题目
思路
如果我们不删除且不设置下界的话就是一道平衡树板子了,那么难点其实就在删除和设置下界。
看了几篇题解,大多是2种方法:
- 插入当前下界,把其和比其小的转移到左子树,然后删除左子树。
- 通过枚举前缀的方式一个一个删除。
2种方法各有优劣,前者浪费空间大,后者花费时间较长。
作为在csp中空间爆掉的选手,我个人建议用后者。
其次就是改下界了,那么我们可以用这样的思路:
每个人都加了工资,相当于下界减少了,反之相同。
所以使用一个和来记录增加工资量。
但是会发现后来的人没加工资,所以我们预先把他们的钱抠走增加量,等查询时就加回去。
code:
#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cstdlib>
using namespace std;
struct Treap{
long long l,r;//左右子编号
long long v,d;//v权值(真),d随机值(旋转用)
long long cnt,size;//cnt重复个数,size子树大小
} a[300010];
long long tot,root=1,n,inf=0x7fffffff,rank1,v;
long long New()//新建节点
{
a[++tot].v=v;
a[tot].d=rand();
a[tot].cnt=1;
a[tot].size=1;
return tot;
}
void Up(long long x)
{
a[x].size=a[a[x].l].size+a[a[x].r].size+a[x].cnt;
return;
}
void Build()
{
v=-inf;
root=New();
v=inf;
New();
a[root].r=2;
//避免越界
Up(root);
return;
}
long long GRBV(long long x)//求排名
{
if (x==0) return -1;//未查询到结果
if (v==a[x].v) return a[a[x].l].size+1;//返回排名
if (v<a[x].v) return GRBV(a[x].l);
return GRBV(a[x].r)+a[a[x].l].size+a[x].cnt;//类似二分
}
long long GVBR(long long x)//求值
{
if (x==0) return -1;//未查询到结果
if (a[a[x].l].size>=rank1) return GVBR(a[x].l);//二分
if (a[a[x].l].size+a[x].cnt>=rank1) return a[x].v;//返回结果
rank1-=a[a[x].l].size+a[x].cnt;
return GVBR(a[x].r);//二分
}
void zig(long long &x)//右旋(左节点变父节点)
{
long long y=a[x].l;
a[x].l=a[y].r,a[y].r=x,x=y;
Up(a[x].r),Up(x);
return;
}
void zag(long long &x)//左旋(右节点变父节点)
{
long long y=a[x].r;
a[x].r=a[y].l,a[y].l=x,x=y;
Up(a[x].l),Up(x);
return;
}
void Insert(long long &x)//插入
{
if (x==0)//新建节点
{
x=New();
return;
}
if (v==a[x].v)//已有节点,个数+1
{
a[x].cnt++,Up(x);
return;
}
if (v<a[x].v)//二分
{
Insert(a[x].l);
if (a[x].d<a[a[x].l].d) zig(x);
Up(x);
return;
}
Insert(a[x].r);
if (a[x].d<a[a[x].r].d) zag(x);
Up(x);
return;
}
long long GP()//求前缀
{
long long ans,x=root;
while (x)
{
if (v>a[x].v)
{
ans=a[x].v,x=a[x].r;
}
else x=a[x].l;
}
return ans;
}
long long GN()//求后缀
{
long long ans,x=root;
while (x)
{
if (v<a[x].v)
{
ans=a[x].v,x=a[x].l;
}
else x=a[x].r;
}
return ans;
}
long long x,last=0,ans=0,ss=0,mn;
void Erase(long long &x)//删除
{
if (x==0) return;
if (v==a[x].v)
{
if (a[x].cnt>1)
{
a[x].cnt--,Up(x);
return;//多个,删一个
}
if (a[x].l||a[x].r)//有子节点
{
if (a[x].r==0||a[a[x].l].d>a[a[x].r].d) zig(x),Erase(a[x].r);
else zag(x),Erase(a[x].l);//旋转,然后删除节点(保证平衡)
Up(x);
}
else x=0;
return;
}
if (v<a[x].v)
{
Erase(a[x].l);
}
else
{
Erase(a[x].r);
}//二分
Up(x);
return;
}
int main()
{
//srand(114514);
Build();
cin>>n>>mn;
char o;
while (n--)
{
cin>>o>>x;
if (o=='I')
{
v=x-last;
if (v>=mn)
{
ss++,ans++;
Insert(root);
}
}
if (o=='S')
{
mn+=x;
last-=x;
v=mn;
while (GP()!=-inf)
{
v=GP();
Erase(root);
v=mn;
ans--;
}
}
if (o=='A')
{
last+=x;
mn-=x;
}
if (o=='F')
{
if (ans<x)
{
cout<<-1<<endl;
}
else
{
rank1=ans-x+2;
cout<<GVBR(root)+last<<endl;
}
}
}
cout<<ss-ans<<endl;
return 0;
}