hdoj 2874 Connections between cities 【Tarjan离线LCA】

港控/mmm° 2022-08-13 16:37 166阅读 0赞

题目：[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]: http://acm.hdu.edu.cn/showproblem.php?pid=2874