HDU - 2586 How far away LCA+tanjar离线算法

传送门

题意:求树上的两点的距离,放在一棵树上,而且题目提示只有一条,很明显是这条路线一定会经过他们的最近公共祖先啦,然后就是一个求距离的问题了。如果要采用离线算法的话,就是通过处理所有询问,那么肯定不能单独的求两点之间的距离,我们肯定是要求每一个点到root节点的距离。利用tanjan算法跑一边就可以。

设root为根结点,询问a,b两点距离,a,b的公共祖先为lca点。dis数组为每个点到根节点的距离。

那么ans=dis[a]+dis[b]-2*dis[lca];

具体做法如下

#include<stdio.h>
#include<string.h>
#include<iostream>
#include<algorithm>
#include<math.h>
#include<set>
#include<stack>
#include<vector>
#include<map>
#include<queue>
#define myself i,l,r
#define lson i<<1
#define rson i<<1|1
#define Lson i<<1,l,mid
#define Rson i<<1|1,mid+1,r
#define half (l+r)/2
#define inff 0x3f3f3f3f
#define lowbit(x) x&(-x)
#define me(a,b) memset(a,b,sizeof(a))
#define min4(a,b,c,d) min(min(a,b),min(c,d))
#define min3(x,y,z) min(min(x,y),min(y,z))
typedef long long ll;
using namespace std;
const int maxn=4e5+5;
struct node
{
    int to,p,val,lca;
}edge[maxn],qedge[maxn];
int vis[maxn],d[maxn],head[maxn],qhead[maxn],f[maxn],s[maxn],e[maxn];
int sign,qsign,n,m;
int find(int x)
{
    if(x==f[x]) return x;
    else return f[x]=find(f[x]);
}
void add(int u,int v,int val)
{
    edge[sign]=node{v,head[u],val};
    head[u]=sign++;
}
void qadd(int u,int v)
{
    qedge[qsign]=node{v,qhead[u]};
    qhead[u]=qsign++;
}
void init()
{
    sign=qsign=0;
    for(int i=1;i<=n;i++)
    {
        f[i]=i;
        head[i]=qhead[i]=-1;
        vis[i]=d[i]=0;
    }
}
void dfs(int u)
{
    vis[u]=1;
    for(int i=head[u];i!=-1;i=edge[i].p)
    {
        int v=edge[i].to;
        if(!vis[v])
        {
            d[v]=d[u]+edge[i].val;
            dfs(v);
            f[v]=u;
        }
    }
    for(int i=qhead[u];i!=-1;i=qedge[i].p)
    {
        int v=qedge[i].to;
        if(vis[v])
        {
            qedge[i].lca=find(v);
            if(i%2==0)
                qedge[i+1].lca=qedge[i].lca;
            else
                qedge[i-1].lca=qedge[i].lca;
        }
    }
}
int main()
{
    int t,x,y,z;
    scanf("%d",&t);
    while(t--)
    {
        scanf("%d %d",&n,&m);
        init();
        for(int i=1;i<n;i++)
        {
            scanf("%d %d %d",&x,&y,&z);
            add(x,y,z);
            add(y,x,z);
        }
        for(int i=1;i<=m;i++)
        {
            scanf("%d %d",&x,&y);
            s[i]=x;
            e[i]=y;
            qadd(x,y);
            qadd(y,x);
        }
        dfs(1);
        for(int i=0;i<m;i++)
        {
            printf("%d\n",d[s[i+1]]+d[e[i+1]]-2*d[qedge[2*i].lca]);
        }
    }
    return 0;
}

猜你喜欢

转载自blog.csdn.net/swust5120171204/article/details/88092835