BZOJ 1056: [HAOI2008]排名系统 & BZOJ 1862: [Zjoi2006]GameZ游戏排名系统

题目地址：http://www.lydsy.com/JudgeOnline/problem.php?id=1056 & http://www.lydsy.com/JudgeOnline/problem.php?id=1862

题目大意：维护3种操作：上传得分（删除原得分），查询排名，查询某段区间的排名。

算法讨论：

        Splay模板题。

        需要注意的是“查询某段区间内的排名”操作。

        如果暴力对区间内的每个值进行查询会TLE。

        那么可以联想到Splay操作：可以先把需要查询的区间通过Splay至Root的右儿子的左子树位置，然后对Root的右儿子的左子树进行中序遍历就可以了。

        Splay的题目总是细节比较多。

/*双倍经验哦亲~*/

Code：

/* * Problem:1056 & 1862 * Author:PYC */#include <cstdio>#include <cstdlib>#include <cstring>#include <algorithm>#define rsz(x) (tree[(x)].r?tree[tree[(x)].r].sz:0)#define lsz(x) (tree[(x)].l?tree[tree[(x)].l].sz:0)#define p 10007#define maxn 1000000#define oo 2147483647using namespace std;int n,tot,tot2,root,num,next[maxn],son[maxn],to[maxn],ans[maxn];char s[20],ed[maxn][20];struct node{int key,l,r,fa,id,sz;}tree[maxn];int gethash(char *s){int l=strlen(s),ans=0;for (int i=0;i<l;++i) ans=(ans*26+s[i]-'A')%p;return ans;}void add(int x,char *y){next[++tot]=son[x];son[x]=tot;memcpy(ed[tot],y,strlen(y));to[tot]=++tot2;}void up(int rt){tree[rt].sz=lsz(rt)+rsz(rt)+1;}int min_node(int rt){if (!tree[rt].l) return rt;return min_node(tree[rt].l);}void zig(int rt){int t=tree[rt].l;tree[rt].l=tree[t].r;tree[t].r=rt;tree[t].fa=tree[rt].fa;tree[rt].fa=t;tree[tree[rt].l].fa=rt;if (rt==tree[tree[t].fa].l) tree[tree[t].fa].l=t;else tree[tree[t].fa].r=t;up(rt);up(t);}void zag(int rt){int t=tree[rt].r;tree[rt].r=tree[t].l;tree[t].l=rt;tree[t].fa=tree[rt].fa;tree[rt].fa=t;tree[tree[rt].r].fa=rt;if (rt==tree[tree[t].fa].l) tree[tree[t].fa].l=t;else tree[tree[t].fa].r=t;up(rt);up(t);}void splay(int x,int &rt){int fa=tree[rt].fa;while (tree[x].fa!=fa){int y=tree[x].fa,z=tree[y].fa;if (z==fa) if (x==tree[y].l) zig(y);else zag(y);else if (y==tree[z].l) if (x==tree[y].l) zig(z),zig(y);else zag(y),zig(z);else if (x==tree[y].r) zag(z),zag(y);else zig(y),zag(z);}rt=x;}void insert(int &rt,int y,int id,int last){if (!rt){rt=to[id];tree[rt].key=y;tree[rt].l=tree[rt].r=0;tree[rt].fa=last;tree[rt].id=id;tree[rt].sz=1;splay(rt,root);return;}if (y<=tree[rt].key){insert(tree[rt].l,y,id,rt);return;}if (y>tree[rt].key){insert(tree[rt].r,y,id,rt);return;}}void del(int x){splay(x,root);if (!tree[root].r){tree[tree[root].l].fa=0;root=tree[root].l;}else{splay(min_node(tree[root].r),tree[root].r);tree[tree[root].r].l=tree[root].l;tree[tree[root].l].fa=tree[root].r;tree[tree[root].r].fa=0;root=tree[root].r;up(root);}}int kth(int &rt,int y){if (y==rsz(rt)+1) return rt;if (y<rsz(rt)+1) return kth(tree[rt].r,y);if (y>rsz(rt)+1) return kth(tree[rt].l,y-rsz(rt)-1);}int rank(int x){splay(x,root);return rsz(root)+1;}void prepare(){tree[1].fa=0;tree[1].id=0;tree[1].key=-oo;tree[1].l=0;tree[1].r=2;tree[1].sz=2;tree[2].fa=1;tree[2].id=0;tree[2].key=oo;tree[2].l=0;tree[2].r=0;tree[2].sz=1;root=1;tot2=2;}void travel(int rt){if (!rt) return;travel(tree[rt].r);ans[++num]=tree[rt].id;travel(tree[rt].l);}int main(){scanf("%d",&n);prepare();for (int i=1;i<=n;++i){char ch=getchar();while (ch!='+' && ch!='?') ch=getchar();if (ch=='+'){int x;scanf("%s%d",s,&x);int H=gethash(s),ans=0;for (int j=son[H];j;j=next[j])if (!strcmp(ed[j],s)){ans=j;break;}if (!ans){add(H,s);insert(root,x,tot,0);}else{del(to[ans]);insert(root,x,ans,0);}}else{scanf("%s",s);if (s[0]>='0' && s[0]<='9'){int l=strlen(s),x=0;for (int j=0;j<l;++j) x=x*10+s[j]-'0';int y=min(x+9,tree[root].sz-2);x++,y++;x=kth(root,x-1);y=kth(root,y+1);swap(x,y);splay(x,root);splay(y,tree[root].r);num=0;travel(tree[tree[root].r].l);for (int i=1;i<num;++i) printf("%s ",ed[ans[i]]);printf("%s\n",ed[ans[num]]);}else{int H=gethash(s);for (int j=son[H];j;j=next[j])if (!strcmp(ed[j],s)){printf("%d\n",rank(to[j])-1);break;}}}}return 0;}

By Charlie Pan

Mar 4,2014