1. 程式人生 > >POJ 2104 歸併樹

POJ 2104 歸併樹

這道題有很多做法,很多人都是用主席樹的板子寫的,這兩天剛好學了一下歸併樹,就決定用歸併樹來過一下這道題。和普通線段樹不同的地方在於,每個節點存的不是一個值,而是一個數組,就是把歸併排序的中間過程儲存下來了。對於每一次查詢,二分已經排好序的陣列,判斷數字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]);
      }
   }
}