POJ 2761 Feed the dogs(Treap名次树+离线处理)

POJ 2761 Feed the dogs(Treap名次树+离线处理)

http://poj.org/problem?id=2761

题意:

        给你N个数,然后要你对下面M个查询回答：(L,R,K)。回答第L个数到第R个数之间的第K小数的值是多少。其中任意给定的两个区间[Li，Ri]和[Lj，Rj]之间不存在包含关系。

分析:

        注意：题目说明中说了没有任何两个区间存在包含关系。这句话告诉我们如果将所有查询区间按L从小到大排序的话，相邻两个区间有这样的关系i<j。Li<Lj且Ri<Rj。(想想是不是这样，如果Ri>=Rj，那么i区间就包含j区间了)

        现在我们就可以离线处理所有查询了，从最左边的查询开始处理，插入该区间的所有节点，使得Treap中只包含该查询中的点，然后返回查询结果.接下来对于下一个查询，先删除不在当前查询区间内的所有节点，然后添加当前查询还缺少的点，然后继续返回所有查询结果即可。

AC代码:

#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<algorithm>
using namespace std;
struct Node
{
    Node *ch[2];
    int r,v;//v值就是N个狗的pretty值
    int s;//节点数
    Node(int v):v(v)
    {
        r=rand();
        s=1;
        ch[0]=ch[1]=NULL;
    }
    int cmp(int x)
    {
        if(x==v) return -1;
        return x<v?0:1;
    }
    void maintain()
    {
        s=1;
        if(ch[0]) s+=ch[0]->s;
        if(ch[1]) s+=ch[1]->s;
    }
};
void rotate(Node* &o,int d)
{
    Node *k=o->ch[d^1];
    o->ch[d^1]=k->ch[d];
    k->ch[d]=o;
    o->maintain();
    k->maintain();
    o=k;
}
void insert(Node* &o,int v)
{
    if(o==NULL) o=new Node(v);
    else
    {
        int d= v<o->v?0:1;//这里可以插入v值重复的
        insert(o->ch[d],v);
        if(o->ch[d]->r > o->r)
            rotate(o,d^1);
    }
    o->maintain();
}
void remove(Node* &o,int v)
{
    int d=o->cmp(v);
    if(d==-1)
    {
        Node *u=o;
        if(o->ch[0] && o->ch[1])
        {
            int d2=o->ch[0]->r < o->ch[1]->r ? 0:1;
            rotate(o,d2);
            remove(o->ch[d2],v);
        }
        else
        {
            if(o->ch[0]==NULL) o=o->ch[1];
            else o=o->ch[0];
            delete u;
        }
    }
    else remove(o->ch[d],v);
    if(o) o->maintain();
}
int kth(Node *o,int k)//找到第k小的v值
{
    if(o==NULL || k<=0 || k>o->s) return -1;
    int s= (o->ch[0]==NULL)? 0:o->ch[0]->s;
    if(k==s+1) return o->v;
    else if(k<=s) return kth(o->ch[0],k);
    else return kth(o->ch[1],k-s-1);
}
const int maxn=100000+100;
const int maxm=50000+500;
struct query
{
    int l,r,k;
    int id;
    bool operator<(const query&b)const
    {
        return l<b.l;
    }
}qs[maxm];
int n,m;
int val[maxn];
int ans[maxm];
int main()
{
    Node *root=NULL;
    scanf("%d%d",&n,&m);
    for(int i=1;i<=n;i++) scanf("%d",&val[i]);
    for(int i=1;i<=m;i++)
    {
        scanf("%d%d%d",&qs[i].l,&qs[i].r,&qs[i].k);
        qs[i].id=i;
    }
    sort(qs+1,qs+m+1);
    int L=1,R=1;//初始在Treap中的节点所属区间[L,R),注意半闭半开区间
    for(int i=1;i<=m;i++)
    {
        while(L<qs[i].l)
        {
            if(L<R) remove(root,val[L]);
            L++;
        }
        if(R<L) R=L;
        while(R<=qs[i].r)
        {
            insert(root,val[R]);
            R++;
        }
        ans[qs[i].id]=kth(root,qs[i].k);
    }
    for(int i=1;i<=m;i++) printf("%d\n",ans[i]);
    return 0;
}