[Vijos]P1034 家族

題目

描述

若某個家族人員過於龐大，要判斷兩個是否是親戚，確實還很不容易，現在給出某個親戚關係圖，求任意給出的兩個人是否具有親戚關係。

規定：x和y是親戚，y和z是親戚，那麼x和z也是親戚。如果x,y是親戚，那麼x的親戚都是y的親戚，y的親戚也都是x的親戚。

格式

輸入格式

第一行：三個整數n,m,p，（n<=5000,m<=5000,p<=5000），分別表示有n個人，m個親戚關係，詢問p對親戚關係。

以下m行：每行兩個數Mi，Mj，1<=Mi，Mj<=N，表示Ai和Bi具有親戚關係。

接下來p行：每行兩個數Pi，Pj，詢問Pi和Pj是否具有親戚關係。
輸出格式

P行，每行一個’Yes’或’No’。表示第i個詢問的答案爲“具有”或“不具有”親戚關係。

樣例1

樣例輸入1

6 5 3
1 2
1 5
3 4
5 2
1 3
1 4
2 3
5 6

樣例輸出1

Yes
Yes
No

限制

各點1S

來源

cdwind整理提交

題解

這道是裸的並查集,我實現的也許不太標準,還是貼出來吧.

#include<cstdio>
using namespace std;
const int maxn = 5010;
int n, m, p, fa[maxn];
void init_set()
{
  for(int i = 1; i <= n; i++)
    fa[i] = i;
}
int find_set(int x)
{
  return fa[x] != x ? fa[x] = find_set(fa[x]) : x;
}
void union_set(int a, int b)
{
  int u = find_set(a);
  int v = find_set(b);
  if(u != v) fa[u] = v;
}
int main()
{
  scanf("%d%d%d", &n, &m, &p);
  init_set();
  int a, b;
  for(int i = 1; i <= m; i++)
  {
    scanf("%d%d", &a, &b);
    union_set(a, b);
  }
  for(int i = 1; i <= p; i++)
  {
    scanf("%d%d", &a, &b);
    if(find_set(a) == find_set(b))
      printf("Yes\n");
    else printf("No\n");
  }
  return 0;
}