主席树这个概念应该不陌生吧!恩?不会, 戳这里。
主席树(函数式线段树)用的是函数思想,一个节点开数组用来保存自己的左右节点,这样节省许多不必要的空间,还可以保存许多历史状态。而这里我们用的是主席树的函数思想来实现。
上题:http://acm.hdu.edu.cn/showproblem.php?pid=5444
题目大意:
给你一个序列,第一个数为二叉树根节点,之后每个数往上加节点,且保证左节点小于根节点,且保证右节点大于根节点。且每个节点最多有2个子节点。然后再查询位置,每往左找输出一个E,右找输出W。例如序列2, 1, 4, 3可以生成如下图:
例如查找1,需要往左一次输出E,查找2,不需要搜直接输出,查找3需要向右一次再向左一次,输出WE。
哇!这题好水,不就是二叉树吗?啪啪啪,几分钟码完了, 交一发,嗯,居然RE了,不行,的开大叔组,开成10W,嗯?又RE了。最后一想如果这个数列是1-n,即a[i] = i,那样需要访问到2的1000次方个节点。咕~~(╯﹏╰)b,郁闷呢,然后回想起以前学过的主席树,可以开数组记录该节点的左右儿子,那样岂不是只要访问到n个节点就行。然后就这样AC了。(比赛的时候提交的时候超了了51s,,14:00:51的时候提交的。本来能AC的,TAT)
附上代码:
#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;
const int N = 4000 + 5;
int a[N], b[N],ls[N], rs[N], mx[N];
int n, k, tot, sz, ql, qr, x, q, T;
void update(int o, int l, int r, int p){
int m = (l + r) >> 1;
if(p <= mx[o]){
if(ls[o] == 0){
ls[o] = tot;
mx[tot] = p;
return ;
}
else update(ls[o], l, m, p);
}
else {
if(rs[o] == 0){
rs[o] = tot;
mx[tot] = p;
return;
}
else update(rs[o], m + 1, r, p);
}
}
void query(int o, int l, int r, int k){
if(mx[o] == k)return ;
int m = (l + r) >> 1;
if(k <= mx[o]){
printf("E");
query(ls[o], l, m, k);
}
else{
printf("W");
query(rs[o], m + 1, r, k);
}
}
void work(){
scanf("%d", &x);
query(1, 1, n, x);
puts("");
}
int main(){
scanf("%d", &T);
while(T--){
scanf("%d", &n);
tot = 1;
//Build(rt[0], 1, n);
memset(mx, 0, sizeof(mx));
memset(ls, 0, sizeof(ls));
memset(rs, 0, sizeof(rs));
//for(int i = 1; i <= n; i ++)ls[i] = i << 1, rs[i] = i << 1|1;
//for(int i = 0; i <= 20; i ++)printf("i = %d, rt = %d, ls = %d, rs= %d, mx = %d\n", i, rt[i], ls[i], rs[i], mx[i]);
for(int i = 1; i <= n; i ++){
scanf("%d", a + i);
if(i == 1)mx[1] = a[1];
else update(1, 1, n, a[i]);
tot ++;
}
scanf("%d", &q);
while(q --)work();
}
return 0;
}