題意

給我們一個n表示操作數(shù)量 然后三種操作
push和pop 還有求中位數(shù)的操作
讓我們根據(jù)操作輸出正確的解

分析

用sort排序做 或者 map標(biāo)記法都會超時
考慮更快的方法
如何快速找到給定一串?dāng)?shù)的中位數(shù)
可以去索引 但是需要排序
題目中告訴我們每個元素都小于1e5
那么也就是說 上下界已知
那么求中位數(shù) 也就是求小于等于某個數(shù)的個數(shù)正好為所有數(shù)的一半
那么求小于等于某個數(shù)的個數(shù) 可以用樹狀數(shù)組去記錄
然后查中位數(shù)時 在樹狀數(shù)組中二分小于等于的個數(shù)mid的元素的下界
樹狀數(shù)組可以用來統(tǒng)計小于等于某元素的數(shù)量

code

#include<bits/stdc++.h> using namespace std; /*樹狀數(shù)組中記錄的是小于當(dāng)前元素的數(shù)量 由于樹狀數(shù)組符合單調(diào)性 所以可以二分這個容量的中間值 */ char a[20]; int c[100003]; vector<int>s; void add(int x,int i){while(x<=100000){c[x]+=i;x+=x&(-x);} } int sum(int x){ //統(tǒng)計小于x的數(shù)量 int ans = 0;while(x){ans+=c[x];x-=x&(-x);}return ans; } int find(int md){int l = 0,r = 100000,mid,ans;while(l<=r){//二分查找小于等于mid元素的數(shù)量 mid = l+r>>1;int ss = sum(mid);if(ss>=md)r = mid-1;else l = mid+1;} return l; } int main() {int n;scanf("%d",&n);while(n--){scanf("%s",a);if(a[1]=='o'){if(s.size()==0){puts("Invalid");continue;}printf("%d\n",s[s.size()-1]);add(s[s.size()-1],-1);s.pop_back();}else if(a[1]=='u'){int val;scanf("%d",&val);add(val,1);s.push_back(val); } else{if(s.size()==0){puts("Invalid");continue;}int f = s.size()+1>>1;printf("%d\n",find(f));}}return 0; }

總結(jié)

以上是生活随笔為你收集整理的pat-1057 Stack 树状数组+二分查找的全部內(nèi)容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網(wǎng)站內(nèi)容還不錯，歡迎將生活随笔推薦給好友。