poj2104（主席树）

这是主席树模板题，至于主席树是什么自己百度吧，我这是给自己留一份主席树的板子。。。。

#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <cmath>
#include <cctype>
#include <iostream>
#include <algorithm>
#include <string>
#include <vector>
#include <queue>
#include <map>
#include <set>
#include <sstream>
#include<iomanip>
using namespace std;
typedef long long ll;
typedef unsigned long long ull;
const ll inff = 0x3f3f3f3f3f3f3f3f;
#define FOR(i,a,b) for(int i(a);i<=(b);++i)
#define FOL(i,a,b) for(int i(a);i>=(b);--i)
#define REW(a,b) memset(a,b,sizeof(a))
#define inf int(0x3f3f3f3f)
#define si(a) scanf("%d",&a)
#define sl(a) scanf("%I64d",&a)
#define sd(a) scanf("%lf",&a)
#define ss(a) scanf("%s",a)
#define mod int(1e9+7)
#define lc (d<<1)
#define rc (d<<1|1)
#define P pair<int,int>
#define pi acos(-1)
struct as{
int l,r,sum;}tr[100008*20];
vector<int>g;
int qw,rt[100008],a[100008],n,m;
int getid(int x)
{
    return lower_bound(g.begin(),g.end(),x)-g.begin()+1;
}
void add(int l,int r,int &x,int y,int pos)
{
    tr[++qw]=tr[y],tr[qw].sum++,x=qw;
    if(l==r)  return;
    int mid=(l+r)>>1;
    if(mid>=pos)  add(l,mid,tr[x].l,tr[y].l,pos);
    else   add(mid+1,r,tr[x].r,tr[y].r,pos);
}
int query(int l,int r,int x,int y,int k)
{
    if(l==r)  return l;
    int mid=(l+r)>>1;
    int sum=tr[tr[y].l].sum-tr[tr[x].l].sum;
    if(sum>=k)  return query(l,mid,tr[x].l,tr[y].l,k);
    else   return query(mid+1,r,tr[x].r,tr[y].r,k-sum);
}
int main()
{
    cin.tie(0);
    si(n),si(m);
    FOR(i,1,n)  si(a[i]),g.push_back(a[i]);
    sort(g.begin(),g.end());
    g.erase(unique(g.begin(),g.end()),g.end());
    FOR(i,1,n)
    {
        add(1,n,rt[i],rt[i-1],getid(a[i]));
    }
    int x,y,z;
    FOR(i,1,m)
    {
        si(x),si(y),si(z);
        cout<<g[query(1,n,rt[x-1],rt[y],z)-1]<<endl;
    }
    return 0;
}

猜你喜欢