儿子个数统计2

时间限制：1秒        内存限制：128M

题目描述

求树中每个点的儿子个数，假设节点1为根。

输入描述

第一行，一个整数n，表示树的节点的个数。
接下来的n-1行中，每行有两个整数x,y，表示x和y有一条边 但是不保证x是y的父亲。

输出描述

一行，n个整数，第i个整数为节点i的儿子个数。

样例

输入

5
1 2
4 1
3 1
2 5

输出

3 1 0 0 0

#include<iostream>
#include<algorithm>
#include<cstring>
#include<string>
#include<cstdio>
using namespace std;
int n,m,q,a,b;
int x,y;
const int N=1e6+10;
int head[N*2],vis[N*2],ver[N*2],Next[N*2],tot=-1,cnt;
void Add(int x,int y) {
	ver[++tot]=y;
	Next[tot]=head[x];
	head[x]=tot;
}
void dfs(int pos) {
	vis[pos]=cnt;
	for(int i=head[pos]; i!=-1; i=Next[i]) {
		int ljd=ver[i];
		if(vis[ljd]==0) {
			dfs(ljd);
		}
	}
}

int main() {
	memset(head,-1,sizeof(head));
	cin>>n>>m>>q;
	for(int i=1; i<=m; i++) {
		cin>>x>>y;
		Add(x,y);
		Add(y,x);
	}
	for(int i=1; i<=n; i++) {
		if(vis[i]==0) {
			cnt++;
			dfs(i);
		}
	}
	while(q--){
		cin>>a>>b;
		if(vis[a]==vis[b]){
			cout<<"Yes"<<endl;
		}
		else{
			cout<<"No"<<endl;
		}
	}
	return 0;
}