poj1470 Closest Common Ancestors（LCA离线优化）

http://poj.org/problem?id=1470

题意：给你点和关系，还有m次询问，求某点是最小公共祖先的次数。

思路：整体思路和poj1330相似，但是这题时间卡的相当紧，听别人说是多组询问。解决办法就是用数组存起来，LCA的时候每处理一个点就处理与该点有关的询问，这样遍历完也就处理完所有询问。保存次数我用的结构体，刚开始想复杂了。。还有输入输出格式也比较恶心，处理完这些后还要记得把数组开大点，因为保存询问会保存的数量级为N*N，这里还卡了我几次WA。。

#include <stdio.h>
#include <algorithm>
#include <stdlib.h>
#include <string.h>
#include <iostream>

using namespace std;

typedef __int64 LL;

const int N = 1005;
const int INF = 0x3f3f3f3f;

int pre[N*N], in[N], head[N], pos, x[N*N], y[N*N], n, m;
bool vis[N];

struct node
{
    int from, to, next;
}edge[2*N];

struct Node
{
    int time;
    bool mark;
}node0[N];

void add(int u, int v)
{
    edge[pos] = (struct node){u, v, head[u]};
    head[u] = pos++;
}

int Find(int x)
{
    if(x == pre[x]) return x;
    pre[x] = Find(pre[x]);
    return pre[x];
}

void LCA(int u)
{
    pre[u] = u;
    vis[u] = true;
    for(int i = 1; i <= m; i++)
    {
        if(u == x[i] && vis[y[i]])
        {
            node0[Find(y[i])].time++;
            node0[Find(y[i])].mark = true;
        }
        if(u == y[i] && vis[x[i]])
        {
            node0[Find(x[i])].time++;
            node0[Find(x[i])].mark = true;
        }
    }
    for(int i = head[u]; i != -1; i = edge[i].next)
    {
        if(!vis[edge[i].to])
        {
            LCA(edge[i].to);
            pre[edge[i].to] = u;
        }
    }
}

void init()
{
    pos = 0;
    for(int i = 0; i <= N; i++)
    {
        node0[i].time = 0;
        node0[i].mark = false;
    }
    memset(head, -1, sizeof(head));
    memset(in, 0, sizeof(in));
    memset(vis, false, sizeof(vis));
    for(int i = 1; i <= n; i++)
        pre[i] = i;
}

int main()
{
   // freopen("in.txt", "r", stdin);
    int u, v, num;
    while(~scanf("%d", &n))
    {
        init();
        for(int i = 0; i < n; i++)
        {
            scanf("%d:(%d)", &u, &num);
            for(int j = 0; j < num; j++)
            {
                scanf("%d", &v);
                add(u, v);
                in[v]++;
            }
        }
        int root;
        for(int i = 1; i <= n; i++)
        {
            if(in[i] == 0) root = i;
        }
        scanf("%d", &m);
        for(int i = 1; i <= m; i++)
            scanf(" (%d %d) ", &x[i], &y[i]);
        LCA(root);
        for(int i = 0; i <= N; i++)
        {
            if(node0[i].mark) printf("%d:%d\n", i, node0[i].time);
        }
    }
    return 0;
}