题目链接: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;
}