链接:https://ac.nowcoder.com/acm/contest/6106/L
来源:牛客网
题目描述
小梁变强之后决定建设自己的道馆,她特别喜欢去其他的道馆串门。
但是有些道馆之间没有道路连通,于是小梁想知道自己能不能去她想去的道馆,
你能帮她写一个程序来查询两个道馆之间是否互相存在道路联通吗;
如果存在输出“YES”,反之输出“NO”。
输入描述:
第一行为三个整数N为道馆个数,M为线路条数,T为查询次数(1≤N<1000,1≤M<1000,1≤T<10000)(1 \leq N<1000,1 \leq M<1000, 1 \leq T<10000)(1≤N<1000,1≤M<1000,1≤T<10000)
第二行至第M+1行,每行两个整数,代表两个道馆的编号a,b(1≤a≤1000,1≤b≤1000)a, b(1 \leq a \leq 1000,1 \leq b \leq 1000)a,b(1≤a≤1000,1≤b≤1000),表示这两个道馆之间有道路相连
第M+2行至第M+T+2M \text{+} T+2M+T+2行,每行两个整数,代表查询这两个道馆。
输出描述:
T行,每行对应一个查询,假如查询的道馆之间可以连接则输出YES,否则则输出NO。
示例1
输入
复制4 2 2 1 3 4 3 1 2 3 4
4 2 2 1 3 4 3 1 2 3 4
输出
复制NO YES
NO YES
递推关系
# include <iostream>
# include <algorithm>
# include <cstring>
# include <math.h>
# include <stdio.h>
# include <vector>
# include <map>
using namespace std;
# define ll long long
const int maxn = 1e6+10;
int a[maxn];
int find(int x){
if(a[x] == x)
return x;
else{
return a[x] = find(a[x]);
}
}
void c(int x, int y){
int f1 = find(x);//f1是已知能到x的源头
int f2 = find(y);
if(f1 != f2)
a[f1] = f2;//标注关系,标注源头
}
int main(){
int n,m,T;
cin >> n >> m >> T;
for(int i = 1; i <= n; i++){
a[i] = i;
}
for(int i = 1; i <= m; i++){
int x,y;
cin >> x >> y;
c(x,y);
}
while(T--){
int x, y;
cin >> x >> y;
if(find(x) == find(y)){
cout << "YES" << endl;
}
else{
cout << "NO" << endl;
}
}
return 0;
}