洛谷P1551 亲戚

题目背景

若某个家族人员过于庞大，要判断两个是否是亲戚，确实还很不容易，现在给出某个亲戚关系图，求任意给出的两个人是否具有亲戚关系。

题目描述

规定：xx 和 yy 是亲戚，yy 和 zz 是亲戚，那么 xx 和 zz 也是亲戚。如果 xx，yy 是亲戚，那么 xx 的亲戚都是 yy 的亲戚，yy 的亲戚也都是 xx 的亲戚。

输入格式

第一行：三个整数 n,m,pn,m,p，（n,m,p \le 5000n,m,p≤5000），分别表示有 nn 个人，mm 个亲戚关系，询问 pp 对亲戚关系。

以下 mm 行：每行两个数 M_iMi​，M_jMj​，1 \le M_i,~M_j\le N1≤Mi​, Mj​≤N，表示 M_iMi​ 和 M_jMj​ 具有亲戚关系。

接下来 pp 行：每行两个数 P_i,P_jPi​,Pj​，询问 P_iPi​ 和 P_jPj​ 是否具有亲戚关系。

输出格式

pp 行，每行一个 Yes 或 No。表示第 ii 个询问的答案为“具有”或“不具有”亲戚关系。

输入输出样例

输入 #1

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

输出 #1

Yes
Yes
No

看了好多题解都是用树来做，我觉得完全可以用弗洛伊德算法，但是时间要注意。

什么是弗洛伊德算法(2条消息) 20.python实现图的最短路径-弗洛伊德算法_woshiwjma956的博客-CSDN博客

如果两个人是陌生人，那么它们之间的距离无限远，最大5000个人，那距离就是10000，如果两个人是亲戚或者两个人通过某个人成为了亲戚，那他们两个之间的距离就是0，最后只要判断两个人之间的距离就可以判断是否为亲戚。

n,m,p = map(int,input().split())
lst = [[10000]*n for i in range(n)]         # 每两个陌生人之间的距离

for i in range(m):                          # 输入亲戚，亲戚之间距离为0
    l,r = map(int,input().split())
    lst[l-1][r-1] = 0
    lst[r-1][l-1] = 0

for i in range(n):          # 弗洛伊德算法
    for j in range(n):
        if i != j:
            for k in range(n):
                if j!=k and i!=k:   # 两个人本来就是亲戚或者通过k成为了亲戚
                    lst[i][j] = min(lst[i][j],lst[i][k]+lst[k][j])

for i in range(p):
    l,r = map(int,input().split())
    if lst[l-1][r-1] == 0:
        print('YES')
    else:
        print('NO')