lca离线模板

最新推荐文章于 2019-05-03 23:29:20 发布

yiqzq

最新推荐文章于 2019-05-03 23:29:20 发布

阅读量203

点赞数

分类专栏： ACM_图论

本文链接：https://blog.csdn.net/yiqzq/article/details/79548751

版权

ACM_图论专栏收录该内容

37 篇文章 0 订阅

订阅专栏

#include <cstring>
#include <cstdio>
#include <algorithm>
#include <cstdlib>
#include <cmath>

using namespace std;
const int N=1e5*4;
struct Edge
{
    int cnt,x[N],y[N],z[N],nxt[N],fst[N];
    void set()
    {
        cnt=0;
        memset(x,0,sizeof x);
        memset(y,0,sizeof y);
        memset(z,0,sizeof z);
        memset(nxt,0,sizeof nxt);
        memset(fst,0,sizeof fst);
    }
    void add(int a,int b,int c)
    {
        x[++cnt]=a;
        y[cnt]=b;
        z[cnt]=c;
        nxt[cnt]=fst[a];//nxt[cnt]保存的是之前的fst(a)的值
        fst[a]=cnt;//cnt应该是第几组数据，fst(a)保存的是以a为父节点的数据的序号
    }
} e,q;
int T,n,m,from,dist,to,in[N],rt,fa[N];
long long dis[N],ans[N];
bool vis[N];
void dfs(int rt)
{
    for (int i=e.fst[rt]; i; i=e.nxt[i])
    {
        dis[e.y[i]]=dis[rt]+e.z[i];
        dfs(e.y[i]);
    }
}
int getf(int k)
{
    return fa[k]==k?k:fa[k]=getf(fa[k]);
}
void LCA(int rt)
{
    for (int i=e.fst[rt]; i; i=e.nxt[i])
    {
        LCA(e.y[i]);
//        fa[getf(e.y[i])]=rt;
        int t1=getf(e.y[i]);
        fa[t1]=rt;
    }
    vis[rt]=1;
    for (int i=q.fst[rt]; i; i=q.nxt[i])
        if (vis[q.y[i]])
            ans[q.z[i]]=dis[q.y[i]]+dis[rt]-2*dis[getf(q.y[i])];
}
int main()
{
    q.set(),e.set();
    memset(in,0,sizeof in);
    memset(vis,0,sizeof vis);
    memset(ans,0,sizeof ans);
    scanf("%d",&n);
    for (int i=1; i<n; i++)
        scanf("%d%d%d",&from,&to,&dist),e.add(from,to,dist),in[to]++;
    scanf("%d",&m);
    for (int i=1; i<=m; i++)
        scanf("%d%d",&from,&to),q.add(from,to,i),q.add(to,from,i);
    rt=0;
    for (int i=1; i<=n&&rt==0; i++)
        if (in[i]==0)
            rt=i;
    dis[rt]=0;
    dfs(rt);
    for (int i=1; i<=n; i++)
        fa[i]=i;
    LCA(rt);
    for (int i=1; i<=m; i++)
        printf("%lld\n",ans[i]);
    return 0;
}