題目詳情點擊這里

思路：用std::stack來表示題目中說的棧，現在關鍵問題就是如何找到中位數。

可以用二分答案＋樹狀數組的方法

由于每個元素最大不超過1e5，因此開一個大小為1e5的樹狀數組來存儲不超過x的數有多少個

每次push一個元素，都把以它為下標的樹狀數組位置+1，pop就相當于-1

這樣的話sum(x)的含義就是棧中不超過x的數有多少個。然后就可以用二分的方法把答案確定出來了。

因為中位數x肯定要滿足sum(x) >= N/2 或者sum(x) >= (N+1)/2

以前太菜了，現在看這道題目，感覺不難

代碼：

#include <iostream> #include <cstdio> #include <stack> using namespace std; const int MAX = 1e5; stack<int> stk; int bitree[MAX + 10]; int N; inline int lowbit(int x) {return x&(-x); } void add(int pos,int x) {while(pos <= MAX){bitree[pos] += x;pos += lowbit(pos);} } int sum(int pos) {int res = 0;while(pos > 0){res += bitree[pos];pos -= lowbit(pos);}return res; } bool check(int x,int size) {if(N&2 == 0)return sum(x) >= size/2; else return sum(x) >= (size+1)/2; } int main() {scanf("%d",&N);char ops[20];for(int i = 0;i < N;i++){scanf("%s",ops);if(ops[1] == 'o'){if(stk.empty())puts("Invalid");else{int val = stk.top();stk.pop();add(val,-1);printf("%d\n",val);}}else if(ops[1] == 'u'){int val;scanf("%d",&val);stk.push(val);add(val,1);}else{if(stk.empty()){puts("Invalid");continue;}int l = 0,r = MAX;while(l < r){int m = (l+r)/2;if(check(m,stk.size()))r = m;elsel = m+1;}printf("%d\n",l);}}return 0;}

總結

以上是生活随笔為你收集整理的L3-002 堆栈 树状数组+二分答案的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。