OJ密码岛 #2341. 【模板】树的故事2

题目描述

你在 CSP-J 的赛场上睡着了。醒来发现自己在一个森林里，这个森林里有 n 棵树，这些树之间共有 n-1 条边相连，并且任意两棵树之间总存在一条路径能互相抵达。

在这些树中，除了 1 号树以外，其他树都有一个父亲树。每一个树都有一个深度，1 号树的深度为 1，其他树的深度为到 1 号树的距离加 1。

现在给出这个森林的连边情况，请你对每个询问输出树的深度。

输入格式

输入共 n+q 行。

第一行两个正整数 n,q 表示树的个数和询问的个数。

接下来 n-1 行，每行两个正整数 u[i],v[i] 表示 u[i] 与 v[i] 之间有一条边相连。

接下来 q 行，每行一个正整数 t[i] 表示询问 u[i] 的深度。

输出格式

输出共 q 行。

第 i 行表示第 i 个询问的答案。

样例

输入样例
4 3
1 2
2 3
3 4
1
3
4

输出样例
1
3
4

代码

#include <iostream>
using namespace std;
int n,q,a[100005],x,y,f[100005];
int read()
{
	int f=1,x=0;
	char c=getchar();
	while(c<'0'||c>'9') {if(c=='-') f=-1;c=getchar();}
	while(c>='0'&&c<='9') {x=x*10+c-48;c=getchar();}
	x*=f;
	return x;
}
void print(int n)
{
	if(n>9) print(n/10);
	putchar(n%10+48);
}
int dfs(int x,int cnt)
{
	if(x==1) return cnt;
	if(!f[x]) f[x]=dfs(a[x],1)+cnt;
	return f[x];
}
int main()
{
	n=read();
	q=read();
	for(int i=1;i<n;i++)
	{
		x=read();
		y=read();
		a[y]=x;
	}
	while(q--)
	{
		x=read();
//		while(x-1) {x=a[x];cnt++;}
		print(dfs(x,1));
		cout << endl;
	}
	return 0;
}