POJ2763 Housewife Wind

树链剖分

有两种操作，

1.修改边的权值

2.询问路径权值和

<span style="font-size:18px;">#include<stdio.h>
#include<string.h>
#include<math.h>
#include<queue>
#include<vector>
#include<iostream>
#include<string>
#include<set>
#include<map>
#include<algorithm>
#include<complex>
using namespace std;
#pragma comment(linker, "/STACK:1024000000,1024000000")
#define nn 100100
#define ll long long
#define ULL unsiged long long
#define mod 1000000007
#define inf oxfffffffffff
#define lson l,mid,rt<<1
#define rson mid+1,r,rt<<1|1

struct edge//记录边
{
    int u,v,c;
    void read()
    {
        scanf("%d%d%d",&u,&v,&c);
    }
}e[nn];
struct node
{
    int to,next;
}v[nn<<1];
int head[nn],dep[nn],siz[nn],son[nn],fa[nn],top[nn],id[nn],val[nn];
int sz,num;
//node和head 构建邻接表
//dep[i] i节点的深度
//siz[i] i节点的儿子个数
//son[i] 与i在同一重链上的i的儿子节点
//fa[i] i节点的父亲节点
//top[i] i节点最近的重链父亲节点
//id[i] i节点的DFS序
//val[i] 在线段树的i节点的权值

void add_edg(int x,int y)
{
    v[sz].to=y;
    v[sz].next=head[x];
    head[x]=sz++;
}

void dfs1(int u,int f,int d)//算出dep,siz,son,fa
{
    dep[u]=d;
    siz[u]=1;
    son[u]=0;
    fa[u]=f;
    for(int i=head[u];i!=-1;i=v[i].next)
    {
        int to=v[i].to;
        if(to!=f)
        {
            dfs1(to,u,d+1);
            siz[u]+=siz[to];
            if(siz[son[u]]<siz[to])
                son[u]=to;
        }
    }
}

void dfs2(int u,int tp)//算出top,id
{
    top[u]=tp;
    id[u]=++num;//dfs序
    if(son[u]) dfs2(son[u],tp);
    for(int i=head[u];i!=-1;i=v[i].next)
    {
        int to=v[i].to;
        if(to==fa[u] || to==son[u]) continue;
        dfs2(to,to);
    }
}

struct tree
{
    int l,r,val;
}t[nn<<2];

void pushup(int rt)
{
    t[rt].val=t[rt<<1].val+t[rt<<1|1].val;
}

void build(int l,int r,int rt)
{
    t[rt].l=l;
    t[rt].r=r;
    if(l==r)
    {
        t[rt].val=val[l];
        return;
    }
    int mid=(l+r)>>1;
    build(lson);
    build(rson);
    pushup(rt);
}

void updata(int rt,int v,int val)
{
    if(t[rt].l==t[rt].r)
    {
        t[rt].val=val;
        return ;
    }
    int mid=(t[rt].l+t[rt].r)>>1;
    if(v<=mid) updata(rt<<1,v,val);
    else updata(rt<<1|1,v,val);
    pushup(rt);
}

int query(int rt,int l,int r)
{
    if(l<=t[rt].l && t[rt].r<=r)
    {
        return t[rt].val;
    }
    int mid=(t[rt].l+t[rt].r)>>1;
    int ans=0;
    if(l<=mid) ans+=query(rt<<1,l,r);
    if(mid<r) ans+=query(rt<<1|1,l,r);
    return ans;
}

int find(int u,int v)
{
    int tp1=top[u];
    int tp2=top[v];
    int ans=0;
    while(tp1!=tp2)
    {
        if(dep[tp1]<dep[tp2])
        {
            swap(tp1,tp2);
            swap(u,v);
        }
        ans+=query(1,id[tp1],id[u]);
        u=fa[tp1];
        tp1=top[u];
    }
    if(u==v) return ans;
    if(dep[u]>dep[v]) swap(u,v);
    ans+=query(1,id[son[u]],id[v]);
    return ans;
}

int main()
{
    int n,m,s;
    while(~scanf("%d%d%d",&n,&m,&s))
    {
        sz=0;
        memset(head,-1,sizeof(head));
        son[0]=0;
        for(int i=1;i<n;i++)
        {
            e[i].read();
            add_edg(e[i].u,e[i].v);
            add_edg(e[i].v,e[i].u);
        }
        num=0;
        dfs1(1,0,1);//节点，父亲，深度
        dfs2(1,1);//节点，最近的重链父亲节点
        for(int i=1;i<n;i++)
        {
            if(dep[e[i].u]<dep[e[i].v])
                swap(e[i].u,e[i].v);
            val[id[e[i].u]]=e[i].c;
        }
        build(1,num,1);
        while(m--)
        {
            int a,b,c;
            scanf("%d%d",&a,&b);
            if(a==0)
            {
                printf("%d\n",find(s,b));
                s=b;
            }
            else
            {
                scanf("%d",&c);
                updata(1,id[e[b].u],c);
            }
        }
    }
    return 0;
}
</span>