P1486 [NOI2004] 郁闷的出纳员

题目

题目

思路

如果我们不删除且不设置下界的话就是一道平衡树板子了,那么难点其实就在删除和设置下界。
看了几篇题解,大多是2种方法:

  1. 插入当前下界,把其和比其小的转移到左子树,然后删除左子树。
  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;
}

猜你喜欢

转载自blog.csdn.net/weixin_49843717/article/details/114764785