PAT-A-1057 Stack (30 分) 树状数组+二分 （C++题解）

1057 Stack (30 分)

题目传送门：1057 Stack (30 分)

一、题目大意

模拟栈的操作，增加找栈中中位数的功能

二、解题思路

第一次写树状数组知识点的题，之前一直不懂树状数组，感觉不好理解那个后缀数组，今天看到一篇很好的文章终于搞懂了树状数组，感谢这位大佬的博客：树状数组简单易懂的详解

通过树状数组维护数组，快速得到前缀和，然后就通过常规的二分，来快速找到中位数。

三、AC代码

#include<bits/stdc++.h>
using namespace std;
const int N = 100003;
vector<int>c(N);
stack<int>S;
int lowbit(int x){
	return x & -x;
}
void update(int index, int val){// 更新后缀数组
	for(;index < N;c[index] += val, index += lowbit(index));
}
int getSum(int index)// {求【0，index】的前缀和
	int ret = 0;
	for(; index > 0; ret += c[index], index -= lowbit(index));
	return ret;
}
int getMedian(){// 二分法找中位数
	int n = S.size();
	int k = (n+1)/2;
	int left = 1, right = N;
	while(left < right){
		int mid = (left + right) >> 1;
		int sum = getSum(mid);
		if(sum < k){
			left = mid+1;
		}else{
			right = mid;
		}
	}
	return left;
}
int main(){
	int T;
	cin >> T;
	while(T--){
		string s;
		cin >> s;
		if(s == "Pop"){
			if(S.empty()){
				cout << "Invalid" << endl;
			}else{
				update(S.top(), -1);
				cout << S.top() << endl;
				S.pop();
			}
		}else if(s == "PeekMedian"){
			if(S.empty()){
				cout << "Invalid" << endl;
			}else{
				cout << getMedian() << endl;
			}
		}else{
			int key;
			cin >> key;
			S.push(key);
			update(key, 1);
		}
	}
}