ZOJ 1141 Closest Common Ancestors（LCA）

Closest Common Ancestors

---

Time Limit: 10 Seconds       Memory Limit: 32768 KB

---

Write a program that takes as input a rooted tree and a list of pairs of vertices. For each pair (u,v) the program determines the closest common ancestor of u and v in the tree. The closest common ancestor of two nodes u and v is the node w that is an ancestor of both u and v and has the greatest depth in the tree. A node can be its own ancestor (for example in Figure 1 the ancestors of node 2 are 2 and 5)

The data set starts with the tree description, in the form:

nr_of_vertices
vertex:(nr_of_successors) successor1 successor2 ... successorn
......

where vertices are represented as integers from 1 to n. The tree description is followed by a list of pairs of vertices, in the form:

nr_of_pairs
(u v) (x y) ...

The input contents several data sets (at least one).

Note that white-spaces (tabs, spaces and line breaks) can be used freely in the input.

For each common ancestor the program prints the ancestor and the number of pair for which it is an ancestor. The results are printed on the standard output on separate lines, in to the ascending order of the vertices, in the format: ancestor:times

For example, for the following tree:

the program input and output is:


Input

5
5:(3) 1 4 2
1:(0)
4:(0)
2:(1) 3
3:(0)
6
(1,5) (1,4) (4,2)
(2,3)
(1,3) (4,3)

Output

2:1
5:5

---

Source:  Southeastern Europe 2000

LCA模板。

代码：

#include <stdio.h>
#include <memory.h>

int g[801][2];

int lca(int x, int y)
{
	while(g[x][1] > g[y][1])
		x = g[x][0];
	while(g[y][1] > g[x][1])
		y = g[y][0];
	while(g[x][1])//非根结点 
	{
		if(x==y)   break; 
		x = g[x][0];
		y = g[y][0];
	}
	return x;
}

int main()
{
	int i, j;
	int n, m;
	int x, y;
	char c;
	int ans[801];
	while(scanf ("%d", &n)!=EOF)
	{   
		memset(ans, 0, sizeof(ans));
		memset(g, 0, sizeof(g));
		for (i = 0; i < n; i ++)
		{
			scanf("%d:(%d) ", &x, &m);
			for(j = 0; j < m; j ++)
			{
				scanf("%d", &y); 
				g[y][0] = x;//保存父节点  
				g[y][1] = g[x][1] + 1;//保存深度 
			}    
		}    
		scanf ("%d\n", &m);
		for (i = 0; i < m; i ++)
		{
			scanf("(%d,%d)%c", &x, &y, &c);
			ans[lca(x, y)] ++;
		}
		//按结点的升序输出结果 
		for(i = 1; i <= n; i ++)
			if(ans[i])
				printf ("%d:%d\n", i, ans[i]);
	}
	return 0;
}