HDU2874 Connections between cities（容易超内存，tarjan离线LCA）

题意

给出一个森林，求任意两点之间的距离。若两者不在一棵树上，输出“Not connected”，否则输出两点之间最短距离。

解题

如果在一棵树上，显然最短距离为u–>lca(u,v)–>v路径上权值之和。设d[u]表示节点u到根节点的距离。
此题卡内存，不能用下面方式存lca。

q[i].lca=q[i^1].lca=find(v);

使前向星只存to和next，不存from和lca。lca用下面方式存

lca[i>>1]=find(v);//第i条边和第i^1条边都对应第i>>1个查询。

AC代码

#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;
const int maxn=1e4+7;
const int maxc=1e6+7;
struct edge
{
    int to,next,w;
}e[maxn<<1];
struct query
{
    int to,next;
}q[maxc<<1];
int lca[maxc];
int head[maxn],first[maxn],cnt,tot;
int par[maxn],n,d[maxn];
bool vis[maxn],V[maxn];
void init()
{
    for(int i=0;i<=n;i++)  d[i]=0;
    for(int i=0;i<=n;i++) vis[i]=false;
    for(int i=0;i<=n;i++) V[i]=false;
    for(int i=0;i<=n;i++) par[i]=i;
    for(int i=0;i<=n;i++) head[i]=-1;
    cnt=-1;
    for(int i=0;i<=n;i++) first[i]=-1;
    tot=-1;
}
void add_edge(int u,int v,int w)
{
    e[++cnt].to=v;
    e[cnt].w=w;
    e[cnt].next=head[u];
    head[u]=cnt;
}
void add_query(int u,int v)
{
    q[++tot].to=v;
    q[tot].next=first[u];
    first[u]=tot;
}
int find(int x)
{
    return par[x]==x?x:par[x]=find(par[x]);
}
void unit(int x,int y)
{
    int fx=find(x),fy=find(y);
    if(fx==fy) return ;
    par[fy]=fx;
}
void tarjan(int u)
{
    vis[u]=true;
    V[u]=true;
    for(int i=head[u];i!=-1;i=e[i].next)
    {
        int v=e[i].to;
        if(vis[v]) continue;
        d[v]=d[u]+e[i].w;
        tarjan(v);
        unit(u,v);
    }
    for(int i=first[u];i!=-1;i=q[i].next)
    {
        int v=q[i].to;
        if(!vis[v]) continue;
        lca[i>>1]=d[u]+d[v]-(d[find(v)]<<1);
    }
}

int main()
{
    int m,c;
    while(~scanf("%d%d%d",&n,&m,&c))
    {
        init();
        for(int i=1;i<=m;i++)
        {
            int u,v,w;
            scanf("%d%d%d",&u,&v,&w);
            add_edge(u,v,w);
            add_edge(v,u,w);
        }
        for(int i=0;i<c;i++)
        {
            int u,v;
            scanf("%d%d",&u,&v);
            add_query(u,v);
            add_query(v,u);
        }
        for(int i=0;i<maxc;i++) lca[i]=-1;
        for(int i=1;i<=n;i++)
        {
            if(!V[i])
            {
                memset(vis,0,sizeof(vis));
                tarjan(i);
            }
        }
        for(int i=0;i<c;i++)
        {
            if(lca[i]==-1)
                printf("Not connected\n");
            else
                printf("%d\n",lca[i]);
        }
    }
    return 0;
}