PAT甲级 Stack (30)

这个题要给一个堆栈赋予输出中位数的功能，直接一想这不是快速排序一下就能过的吗nlogn的复杂度，当然要是十万个数据的话。。网上大部分人使用树状数组做的，思路是用A数组各元素下标表示该元素的数量，然后用树状数组给这个虚拟A数组求和，得到最小的，但是和又为1的那个下标，用二分法写就很方便。总体来说这个算法求和乘以二分，是(logn)^2的复杂度，可以说是相当方便了。其中二分法是收敛到左侧小于中位和，而右侧等于中位和的那个数。

#include<iostream>
using namespace std;
int lowbit(int x);
void add(int x,int a);//此处这个add是在第x位上加或者减一
int sum(int x);//这个是求前sum的数值
int find(int x);//找出第x个数
int c[100000]={0};//C数组
int n;//总个数
int main() {
	cin >> n;
	char s[12];
	int stack[100000];
	int top = -1;
	
	for (int i = 0;i < n;i++) {
		cin >> s;
		if (s[1] < 'o') {//indicates it's peekmedium，目前只能想到logn方的算法
			if (top < 0) {
				cout << "Invalid" << endl;
			}
			else {
				cout << find((top + 2) / 2) << endl;
			}
		}
		else if (s[1] == 'o') {//Pop
			if (top <0) {
				cout << "Invalid" << endl;
			}
			else {
				cout << stack[top] << endl;
				add(stack[top--], -1);
			}
		}
		else {//Push
			int t;
			cin >> t;
			if (top == n) {
				cout << "Invalid" << endl;
			}
			else {
				stack[++top] = t;
				add(t,1);
			}
		}
	 }
	return 0;
}
int lowbit(int x) {
	return x&(-x);
}
void add(int x,int a) {//更新C数组
	for (int i = x;i <= 100000;i += lowbit(i)) {
		c[i] += a;
	}
}
int sum(int x) {//求前x个数的和
	int sum = 0;
	for (int i = x;i > 0;i -= lowbit(i)) {
		sum += c[i];
	}
	return sum;
}
int find(int x) {
	int l =0 , r = 100000;
	int mid ;
	while (r - l != 1) {
		mid = (l + r + 1)/2 ;
		if (sum(mid) >= x) {
			r = mid;
		}
		else {
			l = mid;
		}
	}
	return r;
}

树状数组类似的应用还有求逆序数，只要求出来某数前面有多少数，后面有多少，知道这个数的位置，即可求出他的逆序数