POJ 2104 歸併樹
阿新 • • 發佈:2019-02-13
這道題有很多做法,很多人都是用主席樹的板子寫的,這兩天剛好學了一下歸併樹,就決定用歸併樹來過一下這道題。和普通線段樹不同的地方在於,每個節點存的不是一個值,而是一個數組,就是把歸併排序的中間過程儲存下來了。對於每一次查詢,二分已經排好序的陣列,判斷數字x在被查詢的區間中有幾個數字比他小。query函式的區間查詢操作基本跟線段樹的查詢操作沒有區別了,就是用upper_bound來判斷在當前區間小於x的數的個數。
#include <iostream> #include <cstdio> #include <cstring> #include <algorithm> #include <vector> using namespace std; const int N = 100000 +5; int t[N],n,q; vector<int>v[N<<2]; void build(int l,int r,int d) { if(l == r) { v[d].clear(); v[d].push_back(t[l]); return; } int m = (l+r) >>1; int ld = d<<1; int rd = d<<1|1; build(l,m,ld); build(m+1,r,rd); v[d].resize(r - l +1); merge(v[ld].begin(),v[ld].end(),v[rd].begin(),v[rd].end(),v[d].begin()); } int query(int ql,int qr,int l,int r,int x,int d) { if(qr < l || r < ql) return 0; if(ql <= l && r <= qr) { return upper_bound(v[d].begin(),v[d].end(),x) - v[d].begin(); } int m = (l+r) >>1; int lc = query(ql,qr,l,m,x,d<<1); int rc = query(ql,qr,m+1,r,x,d<<1|1); return lc + rc; } int main() { while(~scanf("%d%d",&n,&q)) { for(int i = 1; i <= n; i++) scanf("%d",&t[i]); build(1,n,1); while(q --) { int a,b,k; scanf("%d%d%d",&a,&b,&k); int l = -1, r = n - 1; while(r - l > 1) { int m = (l + r) >> 1; int c = query(a,b,1,n,v[1][m],1); if(c >= k) r = m; else l = m; } printf("%d\n",v[1][r]); } } }