GFOJ problem468 出去玩 解题报告

题目：http://www.gdfzoj.com/oj/contest/105/problems/3
给出一棵小于n个节点的树，和m个询问，对于每个询问，有两个点，求点a到点b的最短距离，其中 n,m < 40000

在这里讲一下倍增LCA算法 虽然说暴力搜索也可以过
f[i][j] 表示 从点i开始，向上2^j个点的位置，由于这道题带权，再用v[i][j] 用相同的方法表示权值，再记录深度d[i]。
于是就有了递推式

f[i][j] = f[f[i][j-1]][j-1];
v[i][j] = v[i][j-1] + v[f[i][j-1]][j-1];
d[i] = d[fa[i][0]] + 1;

对于每次询问，先用 lowbit 使 a ,b 深度相等，
再用二进制枚举使 a,b 有公共祖先的最小向上深度每一位的值。

再说实现，一开始用 vector 记录邻接表 RE，后来手写邻接表AC……

下面放代码：

#include <cstdio>
#define lb(x) (x & (-x))

int n,q,i,j,k,f[50000][20],v[50000][20],d[50000],a,b,dep,ans,head[50000],last[50000],tot;
struct E {int t,v,next;} e[100000];

void dfs(int u,int l) {
    d[u] = d[l] + 1;
    for (int i=head[u];i;i=e[i].next) if (e[i].t == l) {
        f[u][0] = l;
        v[u][0] = e[i].v;
        for (j=1;j<18;j++) {
            f[u][j] = f[f[u][j-1]][j-1];
            v[u][j] = v[u][j-1] + v[f[u][j-1]][j-1];
        }
    } for (int i=head[u];i;i=e[i].next) if (e[i].t != l) dfs(e[i].t,u);
}

int main() {
    scanf("%d%d",&n,&q);
    for (i=1;i<n;i++) {
        scanf("%d%d%d",&a,&b,&k);
        e[++tot].t = b;  e[tot].v = k;
        last[a] = last[a] ? e[last[a]].next = tot : head[a] = tot;
        e[++tot].t = a;  e[tot].v = k;
        last[b] = last[b] ? e[last[b]].next = tot : head[b] = tot;
    }

    for (dfs(1,0);q--;ans = 0) {
        scanf("%d%d",&a,&b);
        if (d[a] > d[b]) {int t=a;  a=b;  b=t;}
        for (dep = d[b] - d[a];dep;dep -= lb(dep)) {
            for (i=0;i<18;i++) if ((1 << i) == lb(dep)) break;
            ans += v[b][i];
            b = f[b][i];
        }
        for (i=17;i>=0;i--) {
            if (f[a][i] != f[b][i]) {
                ans += v[a][i] + v[b][i];
                a = f[a][i];
                b = f[b][i];
            }
        }
        if (a != b) {
            ans += v[a][0] + v[b][0];
            a = f[a][0];
            b = f[b][0];
        } printf("%d\n",ans);
    }
}