POJ 2763 Housewife Wind (树链剖分)

题意：

给你n个点的树，每个边有个权值，开始处于点m,现在有q次操作，每次单点修改或者将点移动到u，对于每次移动输出经过路程的权值

思路：

树链剖分的基础题，不同点在于每次记得更新点时候把点的更新即可

错误及反思：

C++跑了1700，G++跑了2200，hdu上faq说过G++处理大量输入的时候会

代码：

#include<cstdio>
#include<algorithm>
#include<cstring>
using namespace std;
#define lson l,m,rt<<1
#define rson m+1,r,rt<<1|1
#define fi first
#define se second
const int N =100010;
int segtree[N*4];
int top[N],si[N],fa[N],fi[N],son[N],depth[N],id[N],val[N];
pair<int,int> e[N*2],b[N];
int m,tot=0,tid=0,n,q;
void addedge(int x,int y)
{
    e[tot].fi=y;
    e[tot].se=fi[x];
    fi[x]=tot++;
    e[tot].fi=x;
    e[tot].se=fi[y];
    fi[y]=tot++;
}
void dfs1(int now,int bef,int dep)
{
    fa[now]=bef;
    depth[now]=dep;
    si[now]=1;
    for(int i=fi[now];i!=-1;i=e[i].se)
        if(e[i].fi!=bef)
        {
            dfs1(e[i].fi,now,dep+1);
            si[now]+=si[e[i].fi];
            if(son[now]==-1) son[now]=e[i].fi;
            else son[now]=si[e[i].fi]>si[son[now]]?e[i].fi:son[now];
        }
}
void dfs2(int now,int tp)
{
    top[now]=tp;
    id[now]=tid++;
    if(son[now]!=-1) dfs2(son[now],tp);
    for(int i=fi[now];i!=-1;i=e[i].se)
        if(e[i].fi!=fa[now]&&e[i].fi!=son[now])
            dfs2(e[i].fi,e[i].fi);
}
void change(int pos,int v,int l,int r,int rt)
{
    if(pos==l&&l==r)
    {
        segtree[rt]=v;
        return;
    }
    int m=(l+r)/2;
    if(m>=pos) change(pos,v,lson);
    else change(pos,v,rson);
    segtree[rt]=segtree[rt<<1]+segtree[rt<<1|1];
}
int query(int L,int R,int l,int r,int rt)
{
    if(L<=l&&R>=r)
        return segtree[rt];
    int m=(l+r)/2;
    int ans=0;
    if(m>=L) ans+=query(L,R,lson);
    if(m<R) ans+=query(L,R,rson);
    return ans;
}
int modify(int L,int R)
{
    int f1=top[L],f2=top[R];
    int ans=0;
    while(f1!=f2)
    {
        if(depth[f1]<depth[f2])
        {
            swap(f1,f2);
            swap(L,R);
        }
        ans+=query(id[f1],id[L],0,tid-1,1);
        L=fa[f1];
        f1=top[L];
    }
    if(L==R) return ans;
    if(depth[L]>depth[R]) swap(L,R);
    ans+=query(id[son[L]],id[R],0,tid-1,1);
    return ans;
}
int main()
{
    memset(first,-1,sizeof(first));
    memset(son,-1,sizeof(son));
    scanf("%d%d%d",&n,&q,&m);
    for(int i=0;i<n-1;i++)
    {
        scanf("%d%d%d",&b[i].fi,&b[i].se,&val[i]);
        addedge(b[i].fi,b[i].se);
    }
    dfs1(1,1,1);
    dfs2(1,1);
    for(int i=0;i<n-1;i++)
    {
        if(depth[b[i].fi]<depth[b[i].se])
            swap(b[i].fi,b[i].se);
        change(id[b[i].fi],val[i],0,tid-1,1);
    }
    while(q--)
    {
        int ta,tb,tc;
        scanf("%d",&ta);
        if(ta)
        {
            scanf("%d%d",&tb,&tc);
            change(id[b[tb-1].fi],tc,0,tid-1,1);
        }
        else
        {
            scanf("%d",&tb);
            printf("%d\n",modify(m,tb));
            m=tb;
        }
    }
}