L3-002 特殊堆栈 (30分)

L3-002 特殊堆栈 (30分)

这题的难点就在于不仅要维护一个堆还要进行排序找中值，排序找中值大家应该很熟悉了，可以用快排，如果是有序的话就可以用二分。 这里我们令第m大的数为中值，这题我们可以用桶排序的思想，用数组的下标，数组的值初始化为0，下标出现了就加1。但这样找中值比较麻烦，每次需要进行求和找出数为m所在的数组下标，这样做肯定超时，这时就需要优化。我们可以用树状数组来进行优化，树状数组一般是求一个区间的和，我们就可以利用它，用这个区间的和来表示数的个数，树状数组sum(i)的值就可以表示1~i区间的数的个数，我们就可以二分下标来找出第m个。

#include<iostream>
#include<string>
#include<stack>
using namespace std;
const int N=100010;
int a[N];
stack<int> s;
int lowbit(int x)
{
	return x&(-x);
}
void add(int x,int y)
{
	for(int i=x;i<N;i+=lowbit(i))
		a[i]+=y; 
}
int sum(int x)
{
	int ans=0;
	for(int i=x;i>0;i-=lowbit(i))
		ans+=a[i];
	return ans;
}
int main()
{
	int n,x;
	cin>>n;
	while(n--)
	{
		string str;
		cin>>str;
		if(str=="Push")
		{
			cin>>x;
			add(x,1);
			s.push(x);
		}
		if(str=="Pop")
		{
			if(s.empty()){
				cout<<"Invalid"<<endl;
			}
			else{
				int y=s.top();s.pop();
				add(y,-1);
				cout<<y<<endl;
			}
		}
		if(str=="PeekMedian")
		{
			if(s.empty()){
				cout<<"Invalid"<<endl;continue;
			} 
			int l=1,r=N,m=int((s.size()+1)/2),ans=N;
			while(l<=r)
			{
				int mid=(l+r)/2;
				if(sum(mid)>=m)
				{
					ans=min(ans,mid);
					r=mid-1;
				}
				else
					l=mid+1;
			}
			cout<<ans<<endl;
		}
	}
}