Connections between cities 【HDU - 2874】【在线LCA算法】

题目链接

---

  昨天刚学了在线LCA，今天就来硬刚这道题还是花了一整天的时间，不过对于LCA却有了更多的理解，这道题在讲述不同根的做法上尤其是很好的。

  题目告诉我们有N个节点和M条边，以及C次询问，每次查询的是【L、R】这两个节点间的距离，还是算得上简单明了的告诉了我们得去用最近公共祖先来做的。

---

  对于LCA，毕竟是初学，还是有些地方容易写出BUG，譬如说这道题，我一开始想把节点放进dfs()中处理，但是这样又如何处理不同根的节点呢？显然是不行的，然后我就换了些许思路，再利用个新的数组去存是否在同一个根里，然后在用root[][]去判断其向上2^J时候能到达的父节点是哪个。

  还有，一开始没想到该如何去处理在同一棵树里的预处理问题（就是预处理算出其向上2^x能到达的父节点），然后，我们用一个vis[]来避免同一棵树上的节点被多次处理，既然已经在树上，就说明不会再连接到其他树上，如果能连接，说明dfs()还能走，所以我们如此处理树的问题。

  还有一个最气的就是初始化了，链式前向星的head[]数组是需要初始化的！！！

---

剩下的坑点就没什么了，这道题还莫名良心的没有卡long long，发现竟然有int也能过的discuss。

写的比较淳朴，第一次嘛：

#include <iostream>
#include <cstdio>
#include <cmath>
#include <string>
#include <cstring>
#include <algorithm>
#include <limits>
#include <vector>
#include <stack>
#include <queue>
#include <set>
#include <map>
#define lowbit(x) ( x&(-x) )
#define INF 0x3f3f3f3f3f3f
#define pi 3.141592653589793
#define e 2.718281828459045
using namespace std;
typedef unsigned long long ull;
typedef long long ll;
const int maxN=10005;
int N, M, C, father[maxN], root[maxN][30], depth[maxN], head[maxN<<1], cnt;
bool vis[maxN];
ll dis[maxN];
void init()
{
    for(int i=1; i<=N; i++) { father[i]=i; dis[i]=0; depth[i]=0; }
    cnt=0;
    memset(root, -1, sizeof(root));
    memset(vis, false, sizeof(vis));
    memset(head, -1, sizeof(head));
}
struct Eddge
{
    int next, to, val;
    Eddge(int a=-1, int b=0, int c=0):next(a), to(b), val(c) {}
}edge[maxN<<1];
void addEddge(int u, int v, int val)
{
    edge[cnt] = Eddge(head[u], v, val);
    head[u] = cnt++;
}
int fid(int x) { return x==father[x]?x:(father[x]=fid(father[x])); }
void mix(int x, int y)
{
    int u=fid(x), v=fid(y);
    if(u != v) father[u] = v;
}
void dfs(int u, int pre)
{
    vis[u] = true;
    for(int i=head[u]; i!=-1; i=edge[i].next)
    {
        int v=edge[i].to, cost=edge[i].val;
        if(v == pre || vis[v]) continue;
        depth[v] = depth[u] + 1;
        dis[v] = dis[u] + cost;
        root[v][0] = u;
        dfs(v, u);
    }
}
void pre_did()
{
    for(int i=1; i<=N; i++)
    {
        if(!vis[i]) dfs(i, -1);
    }
    for(int j=0; (1<<(j+1))<N; j++)
    {
        for(int i=1; i<=N; i++)
        {
            if(root[i][j] < 0) root[i][j+1] = -1;
            else root[i][j+1] = root[root[i][j]][j];
        }
    }
}
int LCA(int u, int v)
{
    if(depth[u] > depth[v]) swap(u, v);
    int det_high = depth[v] - depth[u];
    for(int i=(int)log2(det_high); i>=0; i--)
    {
        if( (1<<i) & det_high ) v = root[v][i];
    }
    if(u == v) return u;
    for(int i=15; i>=0; i--)
    {
        if(root[u][i] != root[v][i])
        {
            u = root[u][i];
            v = root[v][i];
        }
    }
    return root[u][0];
}
int main()
{
    while(scanf("%d%d%d", &N, &M, &C)!=EOF)
    {
        init();
        for(int i=1; i<=M; i++)
        {
            int e1, e2, e3;
            scanf("%d%d%d", &e1, &e2, &e3);
            addEddge(e1, e2, e3);
            addEddge(e2, e1, e3);
            mix(e1, e2);
        }
        pre_did();
        for(int i=1; i<=C; i++)
        {
            int e1, e2;
            scanf("%d%d", &e1, &e2);
            if(fid(e1) != fid(e2)) printf("Not connected\n");
            else printf("%lld\n", dis[e1] + dis[e2] - 2*dis[LCA(e1, e2)]);
        }
    }
    return 0;
}