码迷,mamicode.com
首页 > 其他好文 > 详细

hdu1754(splay)

时间:2015-10-20 20:56:37      阅读:205      评论:0      收藏:0      [点我收藏+]

标签:

给n个数,有两种操作    Q a b   询问区间[a,b]的最大值,  U a b 将第a个数的值改成b

 

splay树的中序遍历是我们所维护的序列。如果要询问区间[a,b]的最大值,那么只要将第a-1个数旋转到根结点, 将第b+1个数旋转到根的右孩子,那么根的右孩子的左子树就是所要查询的区间。我们为每一个结点维护一个最大值,表示该以该结点为根的子树的最大值,  那么答案就是 Max[next[next[root][1]][0]];

为了防止越界, 比如要查询区间[1,n]  那么要将第0个数旋转到根结点,将第n+1个数旋转到根的右孩子,   但是却没有这两个数。 所以为了方便,在序列的两端加上两个数,这两个数比序列中的所有数都小, 所以并不影响答案。

 

  1 #include <iostream>
  2 #include <stdio.h>
  3 #include <string.h>
  4 #include <algorithm>
  5 const int  N = 200000 + 10;
  6 int next[N][2],pre[N],key[N],Max[N],sz[N],tot,root;
  7 int a[N];
  8 
  9 //如果更新第x个结点, 那么将第该结点splay到根,然后更新
 10 /*
 11 如果询问区间[a,b]的最大值, 那么将a-1 splay到root,将b+1旋到next[root][1]
 12 
 13 */
 14 void newNode(int &rt, int fa, int k)
 15 {
 16     rt = ++tot;
 17     next[rt][0] = next[rt][1] = 0;
 18     sz[rt] = 1;
 19     pre[rt] = fa;
 20     key[rt] = k;
 21     Max[rt] = k;//???????
 22 }
 23 void maintain(int rt)
 24 {
 25     Max[rt] = std::max(key[rt], std::max(Max[next[rt][0]],Max[next[rt][1]]));
 26     sz[rt] = sz[next[rt][0]] + sz[next[rt][1]] + 1;
 27 }
 28 void build(int &rt, int fa, int l, int r)
 29 {
 30     if(l>r) return;
 31     int m =(l+r)>>1;
 32     newNode(rt,fa,a[m]);
 33     build(next[rt][0],rt,l,m-1);
 34     build(next[rt][1],rt,m+1,r);
 35     maintain(rt);
 36 }
 37 void rotate(int x, int kind)
 38 {
 39     int y = pre[x];
 40     next[y][!kind] = next[x][kind];
 41     maintain(y);
 42     pre[next[x][kind]] = y;
 43     if(pre[y])
 44         next[pre[y]][next[pre[y]][1]==y] = x;
 45     pre[x] = pre[y];
 46     next[x][kind] = y;
 47     pre[y] = x;
 48     maintain(x);
 49 }
 50 int kth(int x, int k)
 51 {
 52     int tmp = sz[next[x][0]] + 1;
 53     if(tmp==k)
 54         return x;
 55     if(tmp > k)
 56         return kth(next[x][0],k);
 57     return kth(next[x][1],k-tmp);
 58 }
 59 void splay(int x, int goal)
 60 {
 61     /*
 62     只考虑左右旋的splay
 63     while(pre[x]!=goal)
 64     {
 65         if(next[pre[x]][0]==x)
 66             rotate(x,1);
 67         else
 68             rotate(x,0);
 69     }
 70     */
 71     while(pre[x]!=goal)
 72     {
 73         int y = pre[x];
 74         if(pre[y]==goal)
 75         {
 76             if(next[y][0]==x)
 77                 rotate(x,1);
 78             else
 79                 rotate(x,0);
 80         }
 81         else
 82         {
 83              //kind 表示y是父亲的哪个儿子, 0 左,1 右
 84             int kind = next[pre[y]][1]==y;
 85             if(next[y][kind]==x)//共线
 86             {
 87                 rotate(y,!kind);
 88                 rotate(x,!kind);
 89             }
 90             else
 91             {
 92                 rotate(x,kind);
 93                 rotate(x,!kind);
 94             }
 95         }
 96     }
 97     if(goal==0)
 98         root = x;
 99 }
100 int main()
101 {
102     int n,m;
103     char opt[3];
104     int x,y;
105     while(scanf("%d%d",&n,&m)!=EOF)
106     {
107         tot = 0;
108         memset(next,0,sizeof(next));
109         for(int i=1;i<=n;++i)
110             scanf("%d",&a[i]);
111         newNode(root,0,-1);
112         newNode(next[root][1],root,-2);
113         build(next[next[root][1]][0],next[root][1],1,n);
114         maintain(next[root][1]);
115         maintain(root);
116         while(m--)
117         {
118             scanf("%s%d%d",opt,&x,&y);
119             if(opt[0]==Q)
120             {
121                 int tmp = kth(root,x);
122                 splay(tmp,0);
123                 splay(kth(root,y+2),root);
124                 printf("%d\n",Max[next[next[root][1]][0]]);
125             }
126             else
127             {
128                 splay(kth(root,x+1),0);
129                 key[root] = Max[root] = y;
130                 maintain(root);
131             }
132         }
133     }
134     return 0;
135 }

 

hdu1754(splay)

标签:

原文地址:http://www.cnblogs.com/justPassBy/p/4895842.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!