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
, head
, pos, x[N*N], y[N*N], n, m;
bool vis
;

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

struct Node
{
int time;
bool mark;
}node0
;

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;
}