Bzoj1056--Haoi2008排名系统

平衡树练习

还是很水的一道题（雾

反正字符串哈希一下会方便很多，其它都是常规操作也没什么好说的

平衡树写得挺繁杂的，以后可能去想想精简一点的版本，写是难写，但这种题一般写出来了就不会错（说完就PE了。。

代码：

#include<bits/stdc++.h>
#define MAXN 250005
#define MAXM 3005
#define INF 10000000000000LL
#define MOD 1000003
#define LL long long
using namespace std;

inline int Hash(char *s) {
int ret=0;
for(int i=strlen(s)-1;i>0;i--) ret=ret*27+s[i]-'A'+1,ret%=MOD;
return ret;
}
struct elem{
char s[15];int v;
};
vector<elem> table[MOD];
inline int _find(char *s) {
int k=Hash(s);int lim=table[k].size(),ret=0;
for(int i=0;i<lim;i++) {
if(!strcmp(table[k][i].s+1,s+1)) ret=table[k][i].v;
}
return ret;
}

int n,bk[MAXN];
char s[15];

struct Node {
int son[2],par,sz;LL v,r;
}x[MAXN];
struct Treap{
int root,L,R,cnt;
void init() {
root=MAXN-2;L=0;R=1;cnt=0;
x[root].v=INF;x[root].r=INF;
}

int pa,t;
inline void rotate(int num,int p) {
pa=x[num].par;t=x[num].sz;
x[num].sz=x[pa].sz;x[pa].sz-=t-x[x[num].son
].sz;
x[x[pa].par].son[x[x[pa].par].son[L]==pa?L:R]=num;x[num].par=x[pa].par;
x[pa].son[p^1]=x[num].son[p];if(x[num].son[p]) x[x[num].son[p]].par=pa;
x[pa].par=num;x[num].son[p]=pa;
}

void Del(int num) {
while(x[num].son[R]|x[num].son[L])
if(x[x[num].son[R]].r>x[x[num].son[L]].r) rotate(x[num].son[R],L);
else rotate(x[num].son[L],R);
int now=x[num].par;
while(now) {x[now].sz--;now=x[now].par;}
x[x[num].par].son[x[x[num].par].son[L]==num?L:R]=0;
}
void Insert(int num,int v) {
int now=root,pre;
x[num].sz=1;x[num].r=rand()%MOD;x[num].v=v;
while(true) {
pre=v>x[now].v?L:R;x[now].sz++;
if(!x[now].son[pre]) {x[num].par=now;x[now].son[pre]=num;break;}
now=x[now].son[pre];
}
while(x[num].r>x[x[num].par].r)
rotate(num,x[x[num].par].son[L]==num?R:L);
}
void Updata(int num,int v,char *s) {
if(!num) {
int k=Hash(s);elem t;strcpy(t.s,s);t.v=++cnt;
table[k].push_back(t),num=cnt;bk[cnt]=k;
}
Del(num);Insert(num,v);
}

bool Qurey_Treap(int v,int p) {
if(x[root].sz<v) return 0;
int now=x[root].son[R],lim;
while(true) {
if(x[x[now].son[L]].sz==v-1) break;
if(x[x[now].son[L]].sz<v) v-=x[x[now].son[L]].sz+1,now=x[now].son[R];
else now=x[now].son[L];
}
lim=table[bk[now]].size();
for(int i=0;i<lim;i++) if(table[bk[now]][i].v==now) {
if(p) printf(" %s",table[bk[now]][i].s+1);
else printf("%s",table[bk[now]][i].s+1);
return 1;
}
}
inline void Qurey_P(int num) {
int ret=x[x[num].son[L]].sz,now=num;
while(now) {
if(x[x[now].par].son[R]==now) ret+=x[x[x[now].par].son[L]].sz+1;
now=x[now].par;
}
printf("%d\n",ret);
}
inline void Qurey_R(int k) {
for(int i=0;i<10;i++) if(!Qurey_Treap(k+i,i)) break;printf("\n");
}
};

int main() {
srand(2333333);
scanf("%d",&n);
Treap p;p.init();
LL a;
for(int i=1;i<=n;i++) {
scanf("%s",s);
if(s[0]=='+') {scanf("%lld",&a);p.Updata(_find(s),a,s);}
if(s[0]=='?') {
if(s[1]>'9') p.Qurey_P(_find(s));
else {
int len=strlen(s);a=0;
for(int i=1;i<len;i++) a=a*10+s[i]-'0';
p.Qurey_R(a);
}
}
}
return 0;
}