poj1986 (树链剖分+线段树或者LCA+RMQ)

求树上任意两点的距离

#include <cstdio>
#include <cstring>
#include <string>
#include <algorithm>
#include <iostream>
#include <queue>
#define INF 0x3f3f3f3f
using namespace std;
int n,m;
#include <vector>
const int N=80005;
vector<int> q[N];
int cnt,tim=0;
int a[N];
int siz[N],son[N]= {0},fa[N],tid[N],dep[N],top[N];
struct node
{
    int u,v,w;
} e[N];
void dfs1(int u,int pre,int de)
{
    siz[u]=1,dep[u]=de,fa[u]=pre;
    int up=q[u].size();
    for(int i=0;i<q[u].size();i++)
    {
        int v=q[u][i];
        if(v!=pre)
        {
            dfs1(v,u,de+1);
            siz[u]+=siz[v];
            if(!son[u]||siz[son[u]]<siz[v])
                son[u]=v;
        }
    }
}
void dfs2(int u,int tp)
{
    top[u]=tp;
    tid[u]=tim++;
//    rk[tid[u]]=u;
    if(!son[u]) return;
    dfs2(son[u],tp);
    int up=q[u].size();
    for(int i=0;i<up;i++)
    {
        int v=q[u][i];
        if(v!=son[u]&&v!=fa[u])
            dfs2(v,v);
    }
}

#define ls l,mid,rt<<1
#define rs mid+1,r,rt<<1|1
int sum[N<<2];

void up(int rt)
{
    sum[rt]=sum[rt<<1]+sum[rt<<1|1];
}
void build(int l,int r,int rt)
{
    if(r==l)
    {
        sum[rt]=a[l];
        return;
    }
    int mid=(l+r)>>1;
    build(ls);
    build(rs);
    up(rt);
}
int query(int L,int R,int l,int r,int rt)
{
    if(L<=l&&r<=R)  return sum[rt];
    int mid=(l+r)>>1;
    int ret=0;
    if(mid>=L) ret+=query(L,R,ls);
    if(mid<R) ret+=query(L,R,rs);
    return ret;
}
int main()
{
    scanf("%d%d",&n,&m);
    for(int i=0; i<m; i++)
    {
        int u,v,w;
        scanf("%d%d%d%*s",&u,&v,&w);
        e[i].u=u,e[i].v=v,e[i].w=w;
        q[u].push_back(v);
        q[v].push_back(u);
    }

    dfs1(1,1,1);

    dfs2(1,1);

    for(int i=0; i<m; i++)
        if(dep[e[i].u]<dep[e[i].v])
            a[tid[e[i].v]]=e[i].w;
        else a[tid[e[i].u]]=e[i].w;
    build(1,n-1,1);
    int k;
    scanf("%d",&k);
    while(k--)
    {
        int aa,bb;
        scanf("%d%d",&aa,&bb);
        int ans=0;
        while(top[aa]!=top[bb])
        {
            if(dep[top[aa]]<dep[top[bb]]) swap(aa,bb);
            ans+=query(tid[top[aa]],tid[aa],1,n-1,1);
            aa=fa[top[aa]];
        }
        if(dep[aa]>dep[bb]) swap(aa,bb);
        if(aa!=bb) ans+=query(tid[aa]+1,tid[bb],1,n-1,1);
        printf("%d\n",ans);
    }
}