SPOJ375.QTREE树链剖分

时间:2023-03-10 05:13:35
SPOJ375.QTREE树链剖分

题意:一个树,a b c 代表a--b边的权值为c。CHANGE x y  把输入的第x条边的权值改为y,QUERY x y 查询x--y路径上边的权值的最大值。

第一次写树链剖分,其实树链剖分只能说是一种思想。树链剖分  就是 先选择从根节点到叶子节点的最长的路径的权值对应到线段树上,然后从一个子树的根节点到叶子的最长路径的权值对应到线段树上这样直到把所有的点都处理了,然后就是线段树区间查询最值了。

具体可以看这个博客。http://blog.sina.com.cn/s/blog_6974c8b20100zc61.html

 #include <cstdio>
#include <cstdlib>
#include <iostream>
#include <cstring>
#include <algorithm>
using namespace std;
typedef unsigned long long ull;
typedef long long ll;
const int inf = 0x3f3f3f3f;
const double eps = 1e-;
const int maxn = 1e4+;
int top[maxn],fa[maxn],son[maxn],dep[maxn],siz[maxn],seg_tree[maxn<<],head[maxn],pos[maxn];
int edge,tot;
struct
{
int to,next;
}e[maxn<<];
void add(int x,int y)
{
e[edge].to = y;
e[edge].next = head[x];
head[x] = edge++;
} void dfs(int r)
{
siz[r] = ;
son[r] = ;
for (int i = head[r]; i > ; i = e[i].next)
if (fa[r] != e[i].to)
{
dep[e[i].to] = dep[r] + ;
fa[e[i].to] = r;
dfs(e[i].to);
if (siz[e[i].to] > siz[son[r]])
son[r] = e[i].to;
siz[r] += siz[e[i].to];
}
} void build(int r,int f)
{
pos[r] = tot++;
top[r] = f;
if (son[r] > )
build(son[r],top[r]);
for (int i = head[r]; i > ; i = e[i].next)
{
if (fa[r] != e[i].to && son[r] != e[i].to)
build(e[i].to,e[i].to);
}
} void update(int l,int r,int o,int x,int v)
{
if (l == r)
{
seg_tree[o] = v;
return;
}
int mid = (l + r) >> ;
if (x <= mid)
update(l,mid,o<<,x,v);
if (x > mid)
update(mid+,r,o<<|,x,v);
seg_tree[o] = max(seg_tree[o<<],seg_tree[o<<|]);
} int query(int l,int r,int o,int ua,int ub)
{
if (ua <= l && ub >= r)
return seg_tree[o];
int mid = (l + r) >> ;
int t1,t2;
t1 = t2 = ;
if (ua <= mid)
t1 = query(l,mid,o<<,ua,ub);
if (ub > mid)
t2 = query(mid+,r,o<<|,ua,ub);
return max(t1,t2);
} int get_max(int ua,int ub)
{
int f1 = top[ua];
int f2 = top[ub];
int tmp = ;
while (f1 != f2)
{
if (dep[f1] < dep[f2])
swap(f1,f2),swap(ua,ub);
tmp = max(tmp,query(,tot,,pos[f1],pos[ua]));
ua = fa[f1],f1 = top[ua];
}
if (ua == ub)
return tmp;
if (dep[ua] > dep[ub])
swap(ua,ub);
tmp = max(tmp,query(,tot,,pos[son[ua]],pos[ub]));
return tmp;
} int d[maxn][];
void init()
{
int n;
scanf ("%d",&n);
memset(dep,,sizeof(dep));
memset(son,,sizeof(son));
memset(head,,sizeof(head));
memset(fa,,sizeof(fa));
memset(top,,sizeof(top));
int root = (n+)>>;
fa[root] = dep[root] =;
tot = edge = ;
int a,b,c;
for (int i = ; i < n; i++)
{
scanf ("%d%d%d",&a,&b,&c);
add(a,b),add(b,a);
d[i][] = a,d[i][] = b,d[i][] = c;
}
dfs(root);
build(root,root);
tot--;
for (int i = ; i < n; i++)
{
if (dep[d[i][]] < dep[d[i][]])
swap(d[i][],d[i][]);
update(,tot,,pos[d[i][]],d[i][]);
} }
int main(void)
{
#ifndef ONLINE_JUDGE
freopen("in.txt","r",stdin);
#endif // ONLINE_JUDGE
int t;
scanf ("%d",&t);
while (t--)
{
init();
char op[];
while (scanf ("%s",op),op[] != 'D')
{
int x,y;
scanf ("%d%d",&x,&y);
if (op[] == 'C')
update(,tot,,pos[d[x][]],y);
if (op[] == 'Q')
printf("%d\n",get_max(x,y));
}
}
return ;
}