URAL-1329 Galactic History

题目链接：http://acm.timus.ru/problem.aspx?space=1&num=1329

题意：给你一个N，接下来有N行数，每行有两个数a,b, 其中b是a是父节点，如果b的父节点是-1，那么b就是所有数的根。然后再输入一个L，接下来有L行数，每行有两个数p,q, 如果p是q的父节点，输出1，如果q是p的父节点， 就输出2，如果都不是，输出0。

思路：一开始想用parent来写，发现只能记录每个点的根，无法记录中间的点，于是想到树形dp常用的head[u]，用它来记录每个点的子节点。也是比较简单的树形dp。

代码如下：

#include<cstdio>
#include<cstring>
const int maxn=40005;
int head[maxn];
int n,root,e,t;

struct node
{
	int id, l, r;//id和r都是记录每个数字出现的顺序（父节点会比子节点先出现），r记录的是每个点的子树长度 
}p[maxn];  //如果那个数是子节点，则r记录的是它在父节点下面的子树出现的顺序 
struct nn
{
	int v, next;
}edge[maxn];
void  addege(int u, int v)
{
	edge[e].v=v;
	edge[e].next=head[u];
	head[u]=e++;
}

void dfs(int u)
{
	p[u].id= t;
	p[u].l=p[u].r=t++;
	for(int i=head[u]; ~i; i=edge[i].next)
	{
		int v=edge[i].v;
		dfs(v);
		if(p[v].r>p[u].r) 
			p[u].r=p[v].r;
	}
}

int main()
{
	while(~scanf("%d", &n))
	{
		e=0;
		int id, fa;
		memset(head, -1, sizeof(head));
		for(int i=1; i<=n; i++)
		{
			scanf("%d%d", &id, &fa);
			if(fa==-1) 	root=id;
			else addege(fa, id); //注意先输父节点，再输子节点 
		}
		t=0;
		dfs(root);
		int m,a,b;
		scanf("%d", &m);
		while(m--)
		{
			scanf("%d%d", &a, &b);
	//如果a出现的时间在b所达到的子树范围内，并且b出现的时间要比a出现的时间早那么b就是a的父节点 
			if(p[a].id<=p[b].r && p[a].id>=p[b].l) puts("2"); 
	// 如果b出现的时间在a所达到的子树范围内，并且a出现的时间要比b出现的时间早那么a就是b的父节点
			else if(p[b].id<=p[a].r && p[b].id>=p[a].l) puts("1");
			else puts("0");
		}
	}
	return 0;
}

下面是用vector写的代码，思路一模一样：

#include <iostream>
#include <string.h>
#include <vector>
using namespace std;
bool take[40005];
int sx[40005],ex[40005],cnt;
vector<int>vec[40005];
void dfs(int x)
{
	take[x] = 1;
	sx[x] = cnt++;
	int st = vec[x].size();
	for(int i=0;i<st;i++)
	{
		int j = vec[x][i];
		if(!take[j]) dfs(j);
	}
	ex[x] = cnt-1;
}
int main()
{
	int n,q,a,b,root;
	cin>>n;
	for (int i=1; i<=n; i++)
	{
		cin>>a>>b;
		if(b!=-1) vec[b].push_back(a);
		else root = a;
	}
	cnt=0;
	memset(sx,0,sizeof(sx));
	memset(ex,0,sizeof(ex));
	dfs(root);
	cin>>q;
	for(int i=1; i<=q; i++)
	{
		cin>>a>>b;
		if(sx[a]<=sx[b] && ex[a]>=ex[b]) cout<<"1"<<endl;
		else if(sx[a]>=sx[b] && ex[a]<=ex[b]) cout<<"2"<<endl;
		else cout<<"0"<<endl;
	}
	return 0;
}