时间限制: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;
}