时间限制:1秒 内存限制:128M
题目描述
输入若干森林,保证数据合法,请你判断某两个点是否在同一棵树上。
输入描述
第一行两个数n,m和q (0<=m<n<=1000000),(1<=q<=10000)表示当前森林的节点个数为n(节点编号从1开始,n结束),以及m条边,接下来m行,每行x和y,表示x和y节点有连边,注:数据中可能存在重边.。接下来q行,每行两个数a和b,表示询问a和b是否位于同一棵树上。
输出描述
对于q中的每个询问,如果a和b属于同一棵树,则输出Yes,否则输出No。
样例
输入
6 3 2 1 2 1 3 2 1 1 2 1 4
输出
Yes No
#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;
}