SPOJ QTREEQuery on a tree

链接：http://acm.hust.edu.cn/vjudge/contest/28982#problem/I

题意：给定一棵n个节点的树，操作A：修改边权，操作B：询问u->v路径上的最大边权。

分析：树链剖分边权练习题，因为是边权而询问是点我们把边权落在节点上，那么我们在询问时是不能计算lca的那个节点的，我是先求了下lca然后将每次在线段是上找时都处理下要大于lca的编号。

代码：

#include<map>
#include<set>
#include<cmath>
#include<queue>
#include<bitset>
#include<math.h>
#include<vector>
#include<string>
#include<stdio.h>
#include<cstring>
#include<iostream>
#include<algorithm>
#pragma comment(linker, "/STACK:102400000,102400000")
using namespace std;
const int N=10010;
const int mod=998244353;
const int MOD1=1000000007;
const int MOD2=1000000009;
const double EPS=0.00000001;
typedef long long ll;
const ll MOD=1000000007;
const int INF=1000000010;
const ll MAX=1ll<<55;
const double eps=1e-5;
const double inf=~0u>>1;
const double pi=acos(-1.0);
typedef double db;
typedef unsigned long long ull;
int tot,u[N],v[2*N],w[2*N],pre[2*N];
void add(int a,int b,int c) {
    v[tot]=b;w[tot]=c;pre[tot]=u[a];u[a]=tot++;
    v[tot]=a;w[tot]=c;pre[tot]=u[b];u[b]=tot++;
}
int n,c[N],f[N],d[N],g[N],h[N];
int fa[N],dep[N],siz[N],son[N],top[N];
void dfs1(int a,int b,int c) {
    dep[a]=dep[b]+1;son[a]=0;
    fa[a]=b;siz[a]=1;d[c]=a;h[a]=c;
    int i,mx=0;
    for (i=u[a];i!=-1;i=pre[i])
    if (v[i]!=b) {
        dfs1(v[i],a,w[i]);siz[a]+=siz[v[i]];
        if (siz[v[i]]>mx) son[a]=v[i],mx=siz[v[i]];
    }
}
void dfs2(int a,int b) {
    if (son[b]==a) top[a]=top[b];
    else top[a]=a;
    f[a]=++tot;g[tot]=c[h[a]];
    if (son[a]) dfs2(son[a],a);
    for (int i=u[a];i!=-1;i=pre[i])
    if (v[i]!=b&&v[i]!=son[a]) dfs2(v[i],a);
}
char s[10];
int mx[N<<2];
void build(int x,int l,int r) {
    if (l==r) { mx[x]=g[l];return ; }
    int mid=(l+r)>>1;
    build(x<<1,l,mid);build(x<<1|1,mid+1,r);
    mx[x]=max(mx[x<<1],mx[x<<1|1]);
}
int getmax(int x,int l,int r,int L,int R) {
    if (L>R) return 0;
    if (l==L&&r==R) return mx[x];
    int mid=(l+r)>>1;
    if (R<=mid) return getmax(x<<1,l,mid,L,R);
    else if (L>mid) return getmax(x<<1|1,mid+1,r,L,R);
    else return max(getmax(x<<1,l,mid,L,mid),getmax(x<<1|1,mid+1,r,mid+1,R));
}
int lca(int a,int b) {
    int f1=top[a],f2=top[b];
    while (f1!=f2) {
        if (dep[f1]<dep[f2]) swap(a,b),swap(f1,f2);
        a=fa[f1];f1=top[a];
    }
    if (dep[a]<dep[b]) swap(a,b);
    return b;
}
int query(int a,int b) {
    int i,ret=0,f1=top[a],f2=top[b],mi=f[lca(a,b)]+1;
    while (f1!=f2) {
        if (dep[f1]<dep[f2]) swap(a,b),swap(f1,f2);
        ret=max(ret,getmax(1,1,n,max(f[f1],mi),f[a]));
        a=fa[f1];f1=top[a];
    }
    if (dep[a]<dep[b]) swap(a,b);
    return max(ret,getmax(1,1,n,max(f[b],mi),f[a]));
}
void updata(int x,int l,int r,int y,int z) {
    if (l==r) { mx[x]=z;return ; }
    int mid=(l+r)>>1;
    if (y<=mid) updata(x<<1,l,mid,y,z);
    else updata(x<<1|1,mid+1,r,y,z);
    mx[x]=max(mx[x<<1],mx[x<<1|1]);
}
int main()
{
    int a,b,i,t;
    scanf("%d", &t);
    while (t--) {
        scanf("%d", &n);
        tot=0;memset(u,-1,sizeof(u));
        for (i=1;i<n;i++) {
            scanf("%d%d%d", &a, &b, &c[i]);add(a,b,i);
        }
        dfs1(1,0,0);tot=0;dfs2(1,0);
        build(1,1,n);
        while (scanf("%s", s)) {
            if (s[0]=='D') break ;
            scanf("%d%d", &a, &b);
            if (s[0]=='Q') printf("%d\n", query(a,b));
            else updata(1,1,n,f[d[a]],b);
        }
    }
    return 0;
}