bzoj1056/1862 [Zjoi2006]GameZ游戏排名系统

题目链接：1,2

treap恶心题，不多说

#include<algorithm>
#include<iostream>
#include<cstdlib>
#include<cstring>
#include<cstdio>
#include<string>
#include<cmath>
#include<queue>
#include<stack>
#include<map>
#include<set>
#define rre(i,r,l) for(int i=(r);i>=(l);i--)
#define re(i,l,r) for(int i=(l);i<=(r);i++)
#define Clear(a,b) memset(a,b,sizeof(a))
#define inout(x) printf("%d",(x))
#define douin(x) scanf("%lf",&x)
#define strin(x) scanf("%s",(x))
#define LLin(x) scanf("%lld",&x)
#define op operator
#define CSC main
typedef unsigned long long ULL;
typedef const int cint;
typedef long long LL;
using namespace std;
cint mod=985003;int n,root,sz,tot,head[mod+1];
struct data1
{
    int v,t,next;
    char ch[10];
}hash[250001];
struct data2
{
    int l,r,v,s,rr,t;
    char ch[10];
}tp[250001];
void maintain(int k){tp[k].s=tp[tp[k].l].s+tp[tp[k].r].s+1;}
void rotate2(int &k)
{
    int t=tp[k].l;
    tp[k].l=tp[t].r;
    tp[t].r=k;tp[t].s=tp[k].s;
    maintain(k);k=t;
}
void rotate1(int &k)
{
    int t=tp[k].r;
    tp[k].r=tp[t].l;
    tp[t].l=k;
    tp[t].s=tp[k].s;
    maintain(k);k=t;
}
bool com(char *a,char *b)
{
    for(int i=1;i<(int)max(strlen(a),strlen(b));i++)
        if(a[i]!=b[i])return 0;
    return 1;
}
int HASH(char *ch)
{
    int s=0; 
    for(int i=1;i<(int)strlen(ch);i++)
    {
        s*=27;s+=(ch[i]-'A'+1);
        s%=mod;
    }
    return s;
}
void del(int &k,int x,int t)
{
    if(tp[k].v==x)
    {
        if(tp[k].t==t)
        {
            if(tp[k].l*tp[k].r==0)k=tp[k].l+tp[k].r;
            else if(tp[tp[k].l].rr<tp[tp[k].r].rr){rotate2(k);del(k,x,t);}
            else {rotate1(k);del(k,x,t);}
        }
        else if(t>tp[k].t){tp[k].s--;del(tp[k].l,x,t);}
        else {tp[k].s--;del(tp[k].r,x,t);}
    }
    else if(x<tp[k].v){tp[k].s--;del(tp[k].l,x,t);}
    else {tp[k].s--;del(tp[k].r,x,t);}
 }
void add(int &k,char *ch,int x,int t)
{
    if(k==0)
    {
        sz++;k=sz;tp[k].v=x;tp[k].s=1;tp[k].rr=rand();
        memcpy(tp[k].ch,ch,strlen(ch));tp[k].t=t;
        return;
    }
    tp[k].s++;
    if(x<=tp[k].v){add(tp[k].l,ch,x,t);if(tp[tp[k].l].rr<tp[k].rr)rotate2(k);}
    else{add(tp[k].r,ch,x,t);if(tp[tp[k].r].rr<tp[k].rr)rotate1(k);}
 }
void add(char *ch,int x,int t)
{
    int k=HASH(ch);int i=head[k];
    while(i)
    {
        if(com(hash[i].ch,ch))
        {
            del(root,hash[i].v,hash[i].t);
            hash[i].t=t;hash[i].v=x;
            add(root,ch,x,t);
            return;
        }
        i=hash[i].next;
    }
    tot++;
    hash[tot].t=t;hash[tot].v=x;
    memcpy(hash[tot].ch,ch,strlen(ch));
    hash[tot].next=head[k];head[k]=tot;
    add(root,ch,x,t);
}
int hh(char *ch)
{
    int k=HASH(ch);int i=head[k];
    while(i)
    {
        if(com(hash[i].ch,ch))return i;
        i=hash[i].next;
    }return 0;
}
int findrank(int k,int x,int t)
{
    if(k==0)return 0;
    if(tp[k].v==x)
    {
        if(tp[k].t>t)return findrank(tp[k].r,x,t);
        else if(tp[k].t<t)return 1+tp[tp[k].r].s+findrank(tp[k].l,x,t);
        else return tp[tp[k].r].s+1;
    }
    else if(tp[k].v<x)return findrank(tp[k].r,x,t);
    else return 1+tp[tp[k].r].s+findrank(tp[k].l,x,t);
}
void query1(char *ch)
{
    int t=hh(ch);
    printf("%d\n",findrank(root,hash[t].v,hash[t].t));
}
int findwei(int k,int x)
{
    if(tp[tp[k].r].s+1==x)return k;
    else if(x<=tp[tp[k].r].s)return findwei(tp[k].r,x);
    else return findwei(tp[k].l,x-tp[tp[k].r].s-1);
}
void query2(char *ch)
{
    int s=0;
    for(int i=1;i<(int)strlen(ch);i++){s*=10;s+=ch[i]-'0';}
    for(int i=s;i<=tot&&i<=s+9;i++)
    {
        printf("%s",tp[findwei(root,i)].ch+1);
        if(i<tot&&i<s+9)printf(" ");
    }
    printf("\n");
}
int CSC()
{
    scanf("%d",&n);char ch[11];int x;
    for(int i=1;i<=n;i++)
    {
        scanf("%s",ch);
        if(ch[0]=='+'){scanf("%d",&x);add(ch,x,i);}
        else if(ch[1]>='A'&&ch[1]<='Z')query1(ch);
        else query2(ch);
    }
    return 0;
}