Hdu 4547 【最近公共祖先】.cpp

题意：

　　dos命令的cd操作有两个可执行方法

　　cd ..    回到上一级目录

　　cd 当前目录名/b/c/d     当前目录名到某一个子目录下

　　

　　给出你目录的关系，然后给出a b问最少经过几步可以从a走到b..

思路：

　　求出a和b的最近公共祖先,然后分4种情况讨论

　　①. a和b有一个公共祖先c，则用 c时间戳-a的时间戳+1(1步可以直接从c到b)

　　②. a是b的祖先，则只用1步就可以到达b点

　　③. b是a的祖先，则用a的时间戳-b的时间戳

　　④. a和b是同一个点，则答案是0

Tips：

　　求绝对值的函数还是得自己写..

　　用algorithm的就wa了..

　　因为algorithm里面的fabs参数和返回值都是double的..

Code：

#include <stdio.h>
#include <cstring>
#include <string>
#include <iostream>
#include <cmath>
#include <map>
using namespace std;

const int MAXM = 10000010;
const int MAXN = 100010;

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

struct Edge
{
    int s;
    int to;
    int next;
    int lca;
}edge[MAXM], qedge[MAXM];
int head[MAXN], qhead[MAXN];
int tot, qtot;

void add(int s, int u)
{
    edge[tot].to = u;
    edge[tot].next = head[s];
    head[s] = tot++;
}

void qadd(int s, int u)
{
    qedge[qtot].s = s;
    qedge[qtot].to = u;
    qedge[qtot].next = qhead[s];
    qhead[s] = qtot++;
    qedge[qtot].s = u;
    qedge[qtot].to = s;
    qedge[qtot].next = qhead[u];
    qhead[u] = qtot++;
}

bool vis[MAXN];
int dfn[MAXN];
int ti;
void LCA(int u)
{
    f[u] = u;
    vis[u] = true;
    for (int i = head[u]; i != -1; i = edge[i].next)
        if (!vis[edge[i].to]) {
            dfn[edge[i].to] = dfn[u]+1;
            LCA(edge[i].to);
            f[edge[i].to] = u;
        }
    for (int i = qhead[u]; i != -1; i = qedge[i].next) {
        if (vis[qedge[i].to]) {
            qedge[i].lca = find(qedge[i].to);
            qedge[i^1].lca = qedge[i].lca;
        }
    }
}

int abs(int x) {
    return x>0?x:-x;
}

int main()
{
    int in[MAXN];
    int T, n, q;
    int cnt, fa;
    char str1[45], str2[45];
    scanf("%d", &T);
    while (T--) {
        scanf("%d %d", &n, &q);
        map<string, int> M;
        cnt = 1;
        tot = qtot = 0;
        ti = 0;

        memset(head, 0xff, sizeof(head));
        memset(qhead, 0xff, sizeof(qhead));
        for (int i = 0; i <= n; ++i) {
            f[i] = i;
            vis[i] = false;
            dfn[i] = 0;
            in[i] = 0;
        }
        for (int i = 0; i < n-1; ++i) {
            scanf("%s %s", str1, str2);
            if (!M[str1]) M[str1] = cnt++;
            if (!M[str2]) M[str2] = cnt++;
            add(M[str2], M[str1]);
            in[M[str1]]++;
        }
        for (int i = 0; i < q; ++i) {
            scanf("%s %s", str1, str2);
            qadd(M[str1], M[str2]);
        }
        for (int i = 1; i <= n; ++i)
            if (in[i] == 0) {
                fa = i;
                break;
            }
        LCA(fa);

     //   for (int i = 1; i < cnt; ++i)
     //       printf("___%d %d\n", i, dfn[i]);

        for (int i = 0; i < qtot; ++i)
        if (i%2 == 0) {
          //  printf("%d %d___ %d\n", qedge[i].s, qedge[i].to, qedge[i].lca);
         //   printf("%d_____\n", qedge[i].lca);
       //  printf("___%d %d\n", dfn[qedge[i].s], dfn[qedge[i].lca]);
            if (qedge[i].to == qedge[i].s) puts("0");
            else if (qedge[i].lca == qedge[i].to) printf("%d\n", abs(dfn[qedge[i].s]-dfn[qedge[i].to]));
            else if (qedge[i].s == qedge[i].lca) puts("1");
            else printf("%d\n", abs(dfn[qedge[i].s]-dfn[qedge[i].lca])+1);
        }
    }
    return 0;
}

题目链接：http://acm.hdu.edu.cn/showproblem.php?pid=4547