2022天梯赛三月冲刺——PAT (Advanced Level)1057 Stack （树状数组+二分查找+栈模拟）

题目大意：模拟栈的入栈、出栈，遇到PeekMedian命令时，输出栈内元素中位数，当栈中没有元素时，输出Invalid.

算法：

1、据说有分块算法，但我并没有去看，hhh，建议大家学一下，分块和莫队，暴力中的女朋友

2、树状数组算法，我们就详细来说一下树状数组的这个方法吧

详细思路：

数据范围，每次查询，暴力查找栈内元素肯定是不行的，所以我们想到用hash来记录每个数出现的次数。所以说，那么序列第K大，就是求最小的i，使得成立。如果用树状数组来解决hash数组求和的问题，那么这个问题就等价于寻找第一个满足条件的getsum,前缀和的i。

由于hash数组的前缀和是递增的，可以令i=1、r=MAXN，然后在[l,r]范围内进行二分，对当前的mid，是否成立。

内心os：感觉这是一个比较巧的思路吧，统计每个数出现的次数，求和，来找到中值，希望可以记下这个巧妙地算法

#include <cstdio>
#include <iostream>
#include <cstdlib>
#include <cstring>
#include <algorithm>
#include <stack>
using namespace std;
#define N 100005
int n;
stack<int>S;
int has[N];
int lowbit(int x)
{
	return x&(-x);
}
void update(int po,int v)
{
	for(int i = po;i < N;i+=lowbit(i))
	{
		has[i]+=v;
	}
}
int getsum(int po)//直接查询前缀和哈，
{//如果要查询一段区间，需要getsum(po2)-getsum(po1),
//不可以直接写到for循环那里，了解一下原理就懂了  
	int sum=0;
	for(int i = po;i > 0;i-=lowbit(i))
	{
		sum+=has[i];
	}
	return sum;
}
void PeekMedian(int k)
{
	int l=1,r=N-1,mid;
	while(l<r)//注意二分的正确姿势哈！！！ 
	{
		mid=(l+r)/2;
		if(getsum(mid)>=k)
		{
			r=mid;
		}else
		{
			l=mid+1;
		}
	}
	printf("%d\n",l);
}
int main()
{
	scanf("%d",&n);
	while(n--)
	{
		string str;
		cin >> str;
		if(str=="Pop")
		{
			if(S.empty())
			{
				cout << "Invalid" <<endl;
			}else
			{
				//has[S.top()]--;
				update(S.top(),-1);
				cout << S.top() <<endl;
				S.pop();
			}
		}else if(str=="Push")
		{
			int x;
			scanf("%d",&x);
			S.push(x);
			update(S.top(),1);
		}else
		{
			if(S.empty())
			{
				cout << "Invalid" <<endl;
			}else
			{
				int kk=S.size();
				if(kk%2)
				{
					PeekMedian((kk+1)/2);
				}else
				{
					PeekMedian(kk/2);
				}
			}
		}
	}
	return 0 ;
}