求各点间最短路算法（树形）

最新推荐文章于 2024-09-01 16:30:56 发布

huanzhizun

最新推荐文章于 2024-09-01 16:30:56 发布

阅读量1.7k

点赞数 4

分类专栏： LCA 文章标签： ACM 最短路 LCA

本文链接：https://blog.csdn.net/u013665921/article/details/26003445

版权

LCA 专栏收录该内容

4 篇文章 0 订阅

订阅专栏

由于floyd算法为O(n^3)算法，肯定会超时，而spfa和dijistala求的是从源点到其他点的最短路径与任意两点间最短路无关，如果做n编上述算法必然超时，所以应该寻求更快的算法。

建议不会LCA的同学先学学习一下，这是一个非常方便的图论算法。

如果图是一颗树的话，把树画出来，可以看出任意两点a，b间距离d[a][b]=dis[a]+dis[b]-2*dis[k],k是a，b的最近公共祖先。

所以只要一遍spfa，然后运用LCA tarjin离线算法把访问点对的最近公共祖先求出来即可。

下面是hdu2874的代码，也可以用来当模板

#include<stdio.h>
#include<iostream>
#include<algorithm>
#include<string.h>
#include<queue>
#define maxn 10010
using namespace std;
struct pi
{
    int to;
    int cost;
    int next;
}pp[2*maxn];
struct ppi
{
    int to;
    int id;
    int next;
};
ppi pp1[200*maxn];
int head[maxn],tot,vis[maxn],ed,head1[maxn],fa[maxn];
int dis[maxn],use[100*maxn];
void add(int a,int b,int cost)
{
    pp[tot].to=b;
    pp[tot].cost=cost;
    pp[tot].next=head[a];
    head[a]=tot++;
    pp[tot].to=a;
    pp[tot].cost=cost;
    pp[tot].next=head[b];
    head[b]=tot++;
}
void add1(int a,int b,int id)
{
    pp1[ed].to=b;
    pp1[ed].id=id;
    pp1[ed].next=head1[a];
    head1[a]=ed++;
    pp1[ed].to=a;
    pp1[ed].id=id;
    pp1[ed].next=head1[b];
    head1[b]=ed++;
}
int find(int a)
{
    if(fa[a]==a||fa[a]==-1)
        return fa[a]=a;
    return fa[a]=find(fa[a]);
}
void LCA(int v,int pa,int to)
{
    int i,k;
    vis[v]=to;
    fa[v]=v;
    for(i=head1[v];i!=-1;i=pp1[i].next)
    {
        if(vis[pp1[i].to]==to)
        {
            k=find(pp1[i].to);
            use[pp1[i].id]=dis[v]+dis[pp1[i].to]-2*dis[k];
        }
    }
    for(i=head[v];i!=-1;i=pp[i].next)
    {
        if(pp[i].to==pa)
            continue;
        dis[pp[i].to]=dis[v]+pp[i].cost;
        LCA(pp[i].to,v,to);
        fa[pp[i].to]=v;
    }
    return ;
}
int main()
{
    int i,j,n,m,p,k,f,to;
    while(scanf("%d%d%d",&n,&m,&f)!=EOF)
    {
        memset(head,-1,sizeof(head));
        memset(head1,-1,sizeof(head1));
        tot=0;
        ed=0;
        for(i=1;i<=m;i++)
        {
            scanf("%d%d%d",&j,&p,&k);
            add(j,p,k);
        }
        for(i=1;i<=n;i++)
        {
            vis[i]=0;
            fa[i]=-1;
            dis[i]=0;
        }
        for(i=1;i<=f;i++)
        {
            scanf("%d%d",&j,&p);
            add1(j,p,i);
            use[i]=-1;
        }
        to=0;
        for(i=1;i<=n;i++)
        {
            if(!vis[i])
            {
                to++;
                LCA(i,-1,to);
            }
        }
        for(i=1;i<=f;i++)
        {
            if(use[i]<0)
                printf("Not connected\n");
            else
                printf("%d\n",use[i]);
        }
    }
    return 0;
}