首页 > 代码库 > hdoj 2874 Connections between cities 【Tarjan离线LCA】

hdoj 2874 Connections between cities 【Tarjan离线LCA】

题目:hdoj 2874 Connections between cities


题意:战争过后,一些城市毁坏了。意思图不连通,让你求任意两点的距离、


分析:很明显求LCA

但是图不连通,所以我们Tarjan的时候要对每个点进行。然后标记即可。

另外,这个题目卡vector,看来以后要学着用数组模拟邻接表了。


AC代码:

#include <iostream>
#include <cstdio>
#include <cstring>
#include <vector>
using namespace std;
#define N 10010
#define M 10010
#define Q 1000010
int head[N];
struct edge{
    int u,v,w,next;
}e[2*M];
int __head[N];
struct ask{
    int u,v,lca,next;
}ea[2*Q];
int dir[N],fa[N],vis[N];
int k;
inline void add_edge(int u ,int v ,int w ,int &k)
{
    e[k].u = u; e[k].v = v; e[k].w = w;
    e[k].next = head[u]; head[u] = k++;
}
inline void add_ask(int u ,int v ,int &k)
{
    ea[k].u = u; ea[k].v = v; ea[k].lca = -1;
    ea[k].next = __head[u]; __head[u] = k++;
}

int find(int x){
    return  x == fa[x] ? x : fa[x] = find(fa[x]);
}

void Tarjan(int u ,int c)
{
    vis[u] = c; fa[u] = u;
    for(int k=head[u]; k!=-1; k=e[k].next)
        if(!vis[e[k].v])
        {
            int v = e[k].v , w = e[k].w;
            dir[v] = dir[u] + w;
            Tarjan(v,c);
            fa[v] = u;
        }
    for(int k=__head[u]; k!=-1; k=ea[k].next)
        if(vis[ea[k].v] == c)
            ea[k].lca = ea[k^1].lca = find(ea[k].v);
}

void Clear(int n)
{
    memset(head,-1,sizeof(head));
    memset(__head,-1,sizeof(__head));
    memset(vis,0,sizeof(vis));
}
int main()
{
    int n,m,c;
    while(~scanf("%d%d%d",&n,&m,&c))
    {
        Clear(n);
        k = 0;
        for(int i=0;i<m;i++)
        {
            int x,y,z;
            scanf("%d%d%d",&x,&y,&z);
            add_edge(x,y,z,k);
            add_edge(y,x,z,k);
        }
        k = 0;
        for(int i=0;i<c;i++)
        {
            int x,y;
            scanf("%d%d",&x,&y);
            add_ask(x,y,k);
            add_ask(y,x,k);
        }
        int cnt = 0;
        for(int i=1;i<=n;i++)
        {
            if(vis[i]==0){
                dir[i] = 0;
                Tarjan(i,++cnt);
            }
        }
        for(int i=0;i<c;i++)
        {
            int f = 2*i;
            int ans = dir[ea[f].u] + dir[ea[f].v] - 2* dir[ea[f].lca];
            //printf("LCA:%d\n",e[f].lca);
            if(ea[f].lca==-1)
                puts("Not connected");
            else
                printf("%d\n",ans);
        }
    }
    return 0;
}


hdoj 2874 Connections between cities 【Tarjan离线LCA】