初涉平衡树「treap」
treap:一种平衡的二叉搜索树
什么是treap(带旋)
treap=tree+heap,这大家都知道。因为二叉搜索树(BST)非常容易被卡成一条链而影响效率,所以我们需要一种更加平衡的树形结构,从而保持$O(logn)$的优秀复杂度。
那么为什么涉及到heap呢?我们知道因为堆有个非常好的性质,它的高度是$logn$的。于是就有人想,能不能够把二叉搜索树的优点与堆的优点结合起来呢?
接下去就要步入OI界的玄学基础————随机数了。
先不管均摊效率高低,我们把BST的每一个节点都赋值一个随机数,然后从两个值分别来看待这个树形结构:1.节点数值val;2.随机数值rnd。
我们把rnd作为关键字来维护一个堆,同时保持在以val为关键字下的二叉搜索树的性质。那么这样我们就有了一颗高度、查询为logn的二叉搜索树了。
至于为什么这样就可以保证在logn……似乎是可以用期望来保证均摊复杂度的优越的。
嘛,具体怎么样就要看人品了……
乍一看好像不可实现,那么让我们来看一些treap功能的代码段。
1 struct node 2 { 3 int son[2],tot,val,rnd; //tot是以当前节点为根的子树大小,val是数值,rnd是随机值 4 int &operator [](const int a) 5 { 6 return son[a]; //son[0]左儿子节点,son[1]右儿子节点 7 } 8 }a[maxn];
1 inline void insert(int &x, int p) //在以x为根的子树中,插入p 2 { 3 if (!x){ //如果当前为空节点,则新建一个节点 4 x = ++cnt,a[cnt].tot = 1; 5 a[cnt].val = p,a[cnt].rnd = randVal(); 6 return; 7 } 8 a[x].tot++; 9 if (p <= a[x].val){ //由二叉搜索树的性质可得,如果p小于等于当前节点,那么应该插入到左子树 10 insert(a[x][0], p); 11 if (a[a[x][0]].rnd < a[x].rnd) rotate(x, 0); //划重点!根据随机值来维护堆的结构! 12 }else{ 13 insert(a[x][1], p); 14 if (a[a[x][1]].rnd < a[x].rnd) rotate(x, 1); 15 } 16 }
那么insert里的rotate是什么操作?
事实上,这就是treap——以及大部分平衡树的核心——旋转操作。
注意一下BST的特殊性质:左儿子小于等于自身;自身大于右儿子。
如果现在向左子树插入完毕之后,左儿子的rnd比自身要小了,按照堆的操作来说,应该将自身和左儿子互换才对。但是这样子就有可能不满足二叉搜索树的性质了。
不过我们可以把左儿子旋转上去,成为自身的祖先,仔细想想就会发现这种技巧同时保证了二叉搜索树和堆的性质。
以上就是rotate操作的全部。
上面这张图可以助于理解rotate;这里是rotate的代码。(我把左旋右旋写在一起了)
1 inline int rotate(int &x, int p) 2 { 3 int t = a[x][p]; 4 a[x][p] = a[t][1^p],a[t][1^p] = x; 5 change(x), change(t); 6 x = t; 7 }
emmm,还是挺优美的一种操作。
理解了treap的旋转操作之后,其他操作相对来说难度就不大了,我们来看一些板子吧。
参考资料:
treap的题目
luoguP3369 【模板】普通平衡树(Treap/SBT)
题目描述
您需要写一种数据结构(可参考题目标题),来维护一些数,其中需要提供以下操作:
- 插入 xx 数
- 删除 xx 数(若有多个相同的数,因只删除一个)
- 查询 xx 数的排名(排名定义为比当前数小的数的个数 +1+1 。若有多个相同的数,因输出最小的排名)
- 查询排名为 xx 的数
- 求 xx 的前驱(前驱定义为小于 xx ,且最大的数)
- 求 xx 的后继(后继定义为大于 xx ,且最小的数)
输入输出格式
输入格式:
第一行为 n ,表示操作的个数,下面 n 行每行有两个数 opt 和 x , opt 表示操作的序号( 1≤opt≤6 )
输出格式:
对于操作 3,4,5,6每行输出一个数,表示对应答案
说明
时空限制:1000ms,128M
1.n的数据范围: n≤100000
2.每个数的数据范围: [−10^7,10^7]
题目分析
这里就是treap(带旋)的所有操作,那么就直接上板子了。
1 #include<bits/stdc++.h> 2 const int maxn = 100003; 3 4 int n,cnt,root; 5 struct node 6 { 7 int son[2],tot,val,rnd; 8 int &operator [](const int a) 9 { 10 return son[a]; 11 } 12 }a[maxn]; 13 14 int randVal() 15 { 16 static int seed = 3627; 17 return seed = int(seed*901832ll*seed%2147483647); 18 } 19 int read() 20 { 21 char ch = getchar(); 22 int num = 0; 23 bool fl = 0; 24 for (; !isdigit(ch); ch = getchar()) 25 if (ch=='-') fl = 1; 26 for (; isdigit(ch); ch = getchar()) 27 num = (num<<1)+(num<<3)+ch-48; 28 if (fl) num = -num; 29 return num; 30 } 31 inline void change(int x) //为了维护子树大小tot。这个tot在以下的ask和find操作中有重要作用。 32 { 33 a[x].tot = a[a[x][0]].tot+a[a[x][1]].tot+1; 34 } 35 inline int rotate(int &x, int p) //旋转 36 { 37 int t = a[x][p]; 38 a[x][p] = a[t][1^p],a[t][1^p] = x; 39 change(x), change(t); 40 x = t; 41 } 42 inline void insert(int &x, int p) //插入 43 { 44 if (!x){ 45 x = ++cnt,a[cnt].tot = 1; 46 a[cnt].val = p,a[cnt].rnd = randVal(); 47 return; 48 } 49 a[x].tot++; 50 if (p <= a[x].val){ 51 insert(a[x][0], p); 52 if (a[a[x][0]].rnd < a[x].rnd) rotate(x, 0); 53 }else{ 54 insert(a[x][1], p); 55 if (a[a[x][1]].rnd < a[x].rnd) rotate(x, 1); 56 } 57 } 58 void del(int &x, int p) //删除 59 { 60 if (a[x].val == p){ 61 if (a[x][0]*a[x][1]==0){ //如果左儿子空或右儿子为空,那么使自身等于非空的那一个节点。 62 x = a[x][0]+a[x][1]; 63 return; 64 } 65 if (a[a[x][0]].rnd < a[a[x][1]].rnd){ //依靠旋转操作保证堆结构,同时边旋转边删除节点 66 rotate(x, 0); 67 del(a[x][1], p); 68 }else{ 69 rotate(x, 1); 70 del(a[x][0], p); 71 } 72 }else{ 73 if (a[x].val > p) del(a[x][0], p); 74 else del(a[x][1], p); 75 } 76 change(x); 77 } 78 int find(int x, int p) //查询以x为根的子树中,p的排名是多少 79 { 80 if (!x) return 1; 81 if (a[x].val >= p) return find(a[x][0], p); 82 return find(a[x][1], p)+a[a[x][0]].tot+1; 83 } 84 int ask(int x, int p) //查询以x为根的子树中,排名为p的数是什么 85 { 86 if (a[a[x][0]].tot == p-1) return a[x].val; 87 if (a[a[x][0]].tot >= p) return ask(a[x][0], p); 88 return ask(a[x][1], p-a[a[x][0]].tot-1); 89 } 90 int pre(int x, int p) //查询在以x为根的子树中,p的前驱是什么 91 { 92 if (!x) return -2e9; 93 if (a[x].val < p) return std::max(a[x].val, pre(a[x][1], p)); 94 return pre(a[x][0], p); 95 } 96 int suf(int x, int p) //查询在以x为根的子树中,p的后驱是什么 97 { 98 if (!x) return 2e9; 99 if (a[x].val > p) return std::min(a[x].val, suf(a[x][0], p)); 100 return suf(a[x][1], p); 101 } 102 int main() 103 { 104 n = read(); 105 for (int i=1; i<=n; i++) 106 { 107 int tt = read(), x = read(); 108 if (tt==1) insert(root, x); 109 if (tt==2) del(root, x); 110 if (tt==3) printf("%d\n",find(root, x)); 111 if (tt==4) printf("%d\n",ask(root, x)); 112 if (tt==5) printf("%d\n",pre(root, x)); 113 if (tt==6) printf("%d\n",suf(root, x)); 114 } 115 return 0; 116 }
luoguP2286 [HNOI2004]宠物收养场
题目描述
凡凡开了一间宠物收养场。收养场提供两种服务:收养被主人遗弃的宠物和让新的主人领养这些宠物。
每个领养者都希望领养到自己满意的宠物,凡凡根据领养者的要求通过他自己发明的一个特殊的公式,得出该领养者希望领养的宠物的特点值a(a是一个正整数,a<2^31),而他也给每个处在收养场的宠物一个特点值。这样他就能够很方便的处理整个领养宠物的过程了,宠物收养场总是会有两种情况发生:被遗弃的宠物过多或者是想要收养宠物的人太多,而宠物太少。
被遗弃的宠物过多时,假若到来一个领养者,这个领养者希望领养的宠物的特点值为a,那么它将会领养一只目前未被领养的宠物中特点值最接近a的一只宠物。(任何两只宠物的特点值都不可能是相同的,任何两个领养者的希望领养宠物的特点值也不可能是一样的)如果有两只满足要求的宠物,即存在两只宠物他们的特点值分别为a-b和a+b,那么领养者将会领养特点值为a-b的那只宠物。
收养宠物的人过多,假若到来一只被收养的宠物,那么哪个领养者能够领养它呢?能够领养它的领养者,是那个希望被领养宠物的特点值最接近该宠物特点值的领养者,如果该宠物的特点值为a,存在两个领养者他们希望领养宠物的特点值分别为a-b和a+b,那么特点值为a-b的那个领养者将成功领养该宠物。
一个领养者领养了一个特点值为a的宠物,而它本身希望领养的宠物的特点值为b,那么这个领养者的不满意程度为abs(a-b)。
你得到了一年当中,领养者和被收养宠物到来收养所的情况,请你计算所有收养了宠物的领养者的不满意程度的总和。这一年初始时,收养所里面既没有宠物,也没有领养者。
输入输出格式
输入格式:
第一行为一个正整数n,n<=80000,表示一年当中来到收养场的宠物和领养者的总数。接下来的n行,按到来时间的先后顺序描述了一年当中来到收养场的宠物和领养者的情况。每行有两个正整数a, b,其中a=0表示宠物,a=1表示领养者,b表示宠物的特点值或是领养者希望领养宠物的特点值。(同一时间呆在收养所中的,要么全是宠物,要么全是领养者,这些宠物和领养者的个数不会超过10000个)
输出格式:
仅有一个正整数,表示一年当中所有收养了宠物的领养者的不满意程度的总和mod 1000000以后的结果。
题目分析
嗯,好像以前的省选题大多是板子题?
这里用到了求不严格前驱、后驱;插入;删除的基本操作。
然后就是这里的题面的幌子:宠物和领养人的区别。注意到同一时间只可能有宠物或者领养人,并且两种身份的领养规则是一样的。于是我们只需要一颗treap就可以了。
另外注意一下INF的设置,我因为看到题面写着:“(a是一个正整数,a<2^31)”然后就因为INF溢出调了好久……
1 #include<bits/stdc++.h> 2 const long long INF = 2147483647; 3 const long long maxn = 120003; 4 const long long MO = 1000000; 5 6 struct node 7 { 8 long long tot,rnd,val,son[2]; 9 long long &operator [] (long long a) 10 { 11 return son[a]; 12 } 13 }a[maxn]; 14 long long n,root,cnt; 15 long long ans; 16 long long ext[2]; 17 18 inline long long read() 19 { 20 char ch = getchar(); 21 long long num = 0; 22 bool fl = 0; 23 for (; !isdigit(ch); ch = getchar()) 24 if (ch=='-') fl = 1; 25 for (; isdigit(ch); ch = getchar()) 26 num = (num<<1)+(num<<3)+ch-48; 27 if (fl) num = -num; 28 return num; 29 } 30 long long randVal() 31 { 32 static long long seed = 3627; 33 return seed = (seed*901832ll*seed%2147483647)%INF; 34 } 35 inline void change(long long x) 36 { 37 a[x].tot = a[a[x][0]].tot+a[a[x][1]].tot+1; 38 } 39 inline void rotate(long long &x, long long p) 40 { 41 long long t = a[x][p]; 42 a[x][p] = a[t][1^p],a[t][1^p] = x; 43 change(x), change(t); 44 x = t; 45 } 46 inline void insert(long long &x, long long p) 47 { 48 if (!x){ 49 x = ++cnt,a[cnt].tot = 1; 50 a[cnt].val = p,a[cnt].rnd = randVal(); 51 return; 52 } 53 a[x].tot++; 54 if (p <= a[x].val){ 55 insert(a[x][0], p); 56 if (a[a[x][0]].rnd < a[x].rnd) rotate(x, 0); 57 }else{ 58 insert(a[x][1], p); 59 if (a[a[x][1]].rnd < a[x].rnd) rotate(x, 1); 60 } 61 } 62 void del(long long &x, long long p) 63 { 64 if (!x) return; 65 if (a[x].val==p){ 66 if (a[x][0]*a[x][1]==0){ 67 x = a[x][0]+a[x][1]; 68 return; 69 } 70 if (a[a[x][0]].rnd < a[a[x][1]].rnd){ 71 rotate(x, 0); 72 del(a[x][1], p); 73 }else{ 74 rotate(x, 1); 75 del(a[x][0], p); 76 } 77 }else{ 78 if (a[x].val > p) 79 del(a[x][0], p); 80 else del(a[x][1], p); 81 } 82 change(x); 83 } 84 long long pre(long long x, long long p) 85 { 86 if (!x) return -INF; 87 if (a[x].val <= p) return std::max(a[x].val, pre(a[x][1], p)); 88 return pre(a[x][0], p); 89 } 90 long long suf(long long x, long long p) 91 { 92 if (!x) return INF; 93 if (a[x].val >= p) return std::min(a[x].val, suf(a[x][0], p)); 94 return suf(a[x][1], p); 95 } 96 void recongnise(long long x) 97 { 98 long long a = pre(root, x); 99 long long b = suf(root, x); 100 if (b!= INF && abs(b-x)<abs(a-x)){ 101 (ans += 1ll*abs(b-x)) %= MO; 102 del(root, b); 103 }else{ 104 (ans += 1ll*abs(a-x)) %= MO; 105 del(root, a); 106 } 107 } 108 int main() 109 { 110 n = read(); 111 for (int i=1; i<=n; i++) 112 { 113 long long tt = read(), x = read(); 114 if (ext[1^tt]){ 115 ext[1^tt]--; 116 recongnise(x); 117 }else{ 118 ext[tt]++; 119 insert(root, x); 120 } 121 } 122 if (ans < 0) ans = 0; 123 printf("%lld\n",ans); 124 return 0; 125 }
luoguP2234 [HNOI2002]营业额统计
题目描述
Tiger最近被公司升任为营业部经理,他上任后接受公司交给的第一项任务便是统计并分析公司成立以来的营业情况。
Tiger拿出了公司的账本,账本上记录了公司成立以来每天的营业额。分析营业情况是一项相当复杂的工作。由于节假日,大减价或者是其他情况的时候,营业额会出现一定的波动,当然一定的波动是能够接受的,但是在某些时候营业额突变得很高或是很低,这就证明公司此时的经营状况出现了问题。经济管理学上定义了一种最小波动值来衡量这种情况:
当最小波动值越大时,就说明营业情况越不稳定。
而分析整个公司的从成立到现在营业情况是否稳定,只需要把每一天的最小波动值加起来就可以了。你的任务就是编写一个程序帮助Tiger来计算这一个值。
第一天的最小波动值为第一天的营业额。
该天的最小波动值=min{|该天以前某一天的营业额-该天营业额|}。
输入输出格式
输入格式:
输入由文件’turnover.in’读入。
第一行为正整数n(n<=32767) ,表示该公司从成立一直到现在的天数,接下来的n行每行有一个整数ai(|ai|<=1000000) ,表示第i天公司的营业额,可能存在负数。
输出格式:
题目分析
emm……这题只用了pre和suf,权且作为练习写写吧。
1 #include<bits/stdc++.h> 2 const int maxn = 40003; 3 const int INF = 2e9; 4 5 struct node 6 { 7 int rnd,val,son[2],tot; 8 int &operator [] (int a) 9 { 10 return son[a]; 11 } 12 }a[maxn]; 13 int n,ans,root,cnt; 14 15 int read() 16 { 17 char ch = getchar(); 18 int num = 0; 19 bool fl = 0; 20 for (; !isdigit(ch); ch = getchar()) 21 if (ch=='-') fl = 1; 22 for (; isdigit(ch); ch = getchar()) 23 num = (num<<1)+(num<<3)+ch-48; 24 if (fl) num = -num; 25 return num; 26 } 27 int randVal() 28 { 29 static int seed = 362721; 30 return seed = int(seed*305267ll%2147483647); 31 } 32 inline void change(int x) 33 { 34 a[x].tot = a[a[x][0]].tot+a[a[x][1]].tot+1; 35 } 36 void rotate(int &x, int p) 37 { 38 int t = a[x][p]; 39 a[x][p] = a[t][1^p], a[t][1^p] = x; 40 change(x), change(t); 41 x = t; 42 } 43 void insert(int &x, int p) 44 { 45 if (x==0){ 46 x = ++cnt,a[x].val = p; 47 a[x].rnd = randVal(), a[x].tot = 1; 48 return; 49 } 50 a[x].tot++; 51 if (p <= a[x].val){ 52 insert(a[x][0], p); 53 if (a[x].rnd > a[a[x][0]].rnd) rotate(x, 0); 54 }else{ 55 insert(a[x][1], p); 56 if (a[x].rnd > a[a[x][1]].rnd) rotate(x, 1); 57 } 58 } 59 int suf(int x, int p) 60 { 61 if (!x) return INF; 62 if (a[x].val >= p) return std::min(a[x].val, suf(a[x][0], p)); 63 return suf(a[x][1], p); 64 } 65 int pre(int x, int p) 66 { 67 if (!x) return -INF; 68 if (a[x].val <= p) return std::max(a[x].val, pre(a[x][1], p)); 69 return pre(a[x][0], p); 70 } 71 int main() 72 { 73 n = read(); 74 for (int i=1; i<=n; i++) 75 { 76 int x = read(); 77 int a = suf(root, x); 78 int b = pre(root, x); 79 if (cnt) 80 ans += std::min(abs(a-x), abs(b-x)); 81 else ans = x; 82 insert(root, x); 83 } 84 printf("%d\n",ans); 85 return 0; 86 }
P1486 [NOI2004]郁闷的出纳员
题目描述
OIER公司是一家大型专业化软件公司,有着数以万计的员工。作为一名出纳员,我的任务之一便是统计每位员工的工资。这本来是一份不错的工作,但是令人郁闷的是,我们的老板反复无常,经常调整员工的工资。如果他心情好,就可能把每位员工的工资加上一个相同的量。反之,如果心情不好,就可能把他们的工资扣除一个相同的量。我真不知道除了调工资他还做什么其它事情。
工资的频繁调整很让员工反感,尤其是集体扣除工资的时候,一旦某位员工发现自己的工资已经低于了合同规定的工资下界,他就会立刻气愤地离开公司,并且再也不会回来了。每位员工的工资下界都是统一规定的。每当一个人离开公司,我就要从电脑中把他的工资档案删去,同样,每当公司招聘了一位新员工,我就得为他新建一个工资档案。
老板经常到我这边来询问工资情况,他并不问具体某位员工的工资情况,而是问现在工资第k多的员工拿多少工资。每当这时,我就不得不对数万个员工进行一次漫长的排序,然后告诉他答案。
好了,现在你已经对我的工作了解不少了。正如你猜的那样,我想请你编一个工资统计程序。怎么样,不是很困难吧?
如果某个员工的初始工资低于最低工资标准,那么将不计入最后的答案内
输入输出格式
输入格式:
第一行有两个非负整数n和min。n表示下面有多少条命令,min表示工资下界。
接下来的n行,每行表示一条命令。命令可以是以下四种之一:
名称 格式 作用
I命令 I_k 新建一个工资档案,初始工资为k。如果某员工的初始工资低于工资下界,他将立刻离开公司。
A命令 A_k 把每位员工的工资加上k
S命令 S_k 把每位员工的工资扣除k
F命令 F_k 查询第k多的工资
_(下划线)表示一个空格,I命令、A命令、S命令中的k是一个非负整数,F命令中的k是一个正整数。
在初始时,可以认为公司里一个员工也没有。
输出格式:
输出文件的行数为F命令的条数加一。
对于每条F命令,你的程序要输出一行,仅包含一个整数,为当前工资第k多的员工所拿的工资数,如果k大于目前员工的数目,则输出-1。
输出文件的最后一行包含一个整数,为离开公司的员工的总数。
说明
I命令的条数不超过100000
A命令和S命令的总条数不超过100
F命令的条数不超过100000
每次工资调整的调整量不超过1000
新员工的工资不超过100000
题目大意
需要一个数据结构支持一下的操作:
1.增加元素a
2.把现在所有的元素同时加上或减去数k
3.删除小于值value的所有元素
4.查询排名为k的元素
题目分析
那么就基本等同于treap的操作了。不过这里衍生出了treap的另一种删除操作:删除所有小于value的元素。
1 int del(int &x, int p) //将del改为int:返回值是以x为根的子树中,小于p的元素有几个。 2 { 3 if (!x) return 0; 4 if (a[x].val < p) 5 { 6 int tt = a[a[x][0]].tot+1; 7 x = a[x][1]; 8 return tt+del(x, p); 9 }else{ 10 int tt = del(a[x][0], p); 11 a[x].tot -= tt; 12 return tt; 13 } 14 }
如果理解透彻了treap的del操作,这里的删除应该也是不难理解的。
然后还有一个小技巧:把一起增减的工资用一个delta记录下来,避免了冗余操作。
1 #include<bits/stdc++.h> 2 const int maxn = 100003; 3 4 struct node 5 { 6 int son[2],rnd,val,tot; 7 int &operator [](int a) 8 { 9 return son[a]; 10 } 11 }a[maxn]; 12 int n,k,cnt,root,sum,delta; 13 14 int read() 15 { 16 char ch = getchar(); 17 int num = 0; 18 for (; !isdigit(ch); ch = getchar()); 19 for (; isdigit(ch); ch = getchar()) 20 num = (num<<1)+(num<<3)+ch-48; 21 return num; 22 } 23 int randVal() 24 { 25 static int seed = 362721; 26 return seed = (seed*326752ll%2147483647); 27 } 28 void change(int x) 29 { 30 a[x].tot = a[a[x][0]].tot+a[a[x][1]].tot+1; 31 } 32 void rotate(int &x, int p) 33 { 34 int t = a[x][p]; 35 a[x][p] = a[t][1^p], a[t][1^p] = x; 36 change(x), change(t); 37 x = t; 38 } 39 void insert(int &x, int p) 40 { 41 if (!x){ 42 x = ++cnt, a[x].tot = 1; 43 a[x].rnd = randVal(), a[x].val = p; 44 return; 45 } 46 a[x].tot++; 47 if (p <= a[x].val){ 48 insert(a[x][0], p); 49 if (a[x].rnd < a[a[x][0]].rnd) rotate(x, 0); 50 }else{ 51 insert(a[x][1], p); 52 if (a[x].rnd < a[a[x][1]].rnd) rotate(x, 1); 53 } 54 } 55 int del(int &x, int p) 56 { 57 if (!x) return 0; 58 if (a[x].val < p) 59 { 60 int tt = a[a[x][0]].tot+1; 61 x = a[x][1]; 62 return tt+del(x, p); 63 }else{ 64 int tt = del(a[x][0], p); 65 a[x].tot -= tt; 66 return tt; 67 } 68 } 69 int ask(int x, int p) 70 { 71 if (a[a[x][0]].tot == p-1) return a[x].val; 72 if (a[a[x][0]].tot >= p) return ask(a[x][0], p); 73 return ask(a[x][1], p-a[a[x][0]].tot-1); 74 } 75 int main() 76 { 77 n = read(), k = read(); 78 for (int i=1; i<=n; i++) 79 { 80 char ch = getchar(); 81 for (; !isalpha(ch); ch = getchar()); 82 int x = read(); 83 if (ch=='I'){ 84 if (x >= k) 85 insert(root, x-delta); 86 }else if (ch=='A'){ 87 delta += x; 88 }else if (ch=='S'){ 89 delta -= x; 90 sum += del(root, k-delta); 91 }else{ 92 if (x > cnt-sum) printf("-1\n"); 93 else printf("%d\n",ask(root, cnt-sum-x+1)+delta); 94 } 95 } 96 printf("%d\n",sum); 97 return 0; 98 }
P4305 [JLOI2011]不重复数字
题目描述
给出N个数,要求把其中重复的去掉,只保留第一次出现的数。
例如,给出的数为1 2 18 3 3 19 2 3 6 5 4,其中2和3有重复,去除后的结果为1 2 18 3 19 6 5 4。
输入输出格式
输入格式:
输入第一行为正整数T,表示有T组数据。
接下来每组数据包括两行,第一行为正整数N,表示有N个数。第二行为要去重的N个正整数。
输出格式:
对于每组数据,输出一行,为去重后剩下的数字,数字之间用一个空格隔开。
说明
对于30%的数据,1 <= N <= 100,给出的数不大于100,均为非负整数;
对于50%的数据,1 <= N <= 10000,给出的数不大于10000,均为非负整数;
对于100%的数据,1 <= N <= 50000,给出的数在32位有符号整数范围内。 T≤50
题目分析
嗯……这题可以用set直接水过去。不过拿来练习平衡树也是挺好的。
1 #include<bits/stdc++.h> 2 const int maxn = 50003; 3 4 struct node 5 { 6 int rnd,val,son[2]; 7 int &operator [](int a) 8 { 9 return son[a]; 10 } 11 }a[maxn]; 12 int tt,n,cnt,root; 13 14 int read() 15 { 16 char ch = getchar(); 17 int num = 0; 18 bool fl = 0; 19 for (; !isdigit(ch); ch = getchar()) 20 if (ch=='-') fl = 1; 21 for (; isdigit(ch); ch = getchar()) 22 num = (num<<1)+(num<<3)+ch-48; 23 if (fl) num = -num; 24 return num; 25 } 26 int randVal() 27 { 28 static int seed = 3627; 29 return seed = (seed*6565748ll%2147483647); 30 } 31 void rotate(int &x, int p) 32 { 33 int t = a[x][p]; 34 a[x][p] = a[t][1^p], a[t][1^p] = x; 35 x = t; 36 } 37 void insert(int &x, int p) 38 { 39 if (!x){ 40 x = ++cnt; 41 a[x].rnd = randVal(),a[x].val = p; 42 return; 43 } 44 if (p <= a[x].val){ 45 insert(a[x][0], p); 46 if (a[x].rnd > a[a[x][0]].rnd) rotate(x, 0); 47 }else{ 48 insert(a[x][1], p); 49 if (a[x].rnd > a[a[x][1]].rnd) rotate(x, 1); 50 } 51 } 52 bool exist(int x, int p) 53 { 54 if (!x) return 0; 55 if (a[x].val == p) return 1; 56 if (p < a[x].val) return exist(a[x][0], p); 57 return exist(a[x][1], p); 58 } 59 int main() 60 { 61 tt = read(); 62 while (tt--) 63 { 64 n = read(); 65 memset(a, 0, sizeof a); 66 cnt = 0, root = 0; 67 for (int i=1; i<=n; i++) 68 { 69 int x = read(); 70 if (!exist(root, x)){ 71 printf("%d ",x); 72 insert(root, x); 73 } 74 } 75 puts(""); 76 } 77 return 0; 78 }
END