题目描述

w 星球的一个种植园，被分成 m×n 个小格子（东西方向 m 行，南北方向 n 列）。每个格子里种了一株合根植物。

这种植物有个特点，它的根可能会沿着南北或东西方向伸展，从而与另一个格子的植物合成为一体。

如果我们告诉你哪些小格子间出现了连根现象，你能说出这个园中一共有多少株合根植物吗？

输入描述

第一行，两个整数 m,n，用空格分开，表示格子的行数、列数

接下来一行，一个整数 k ，表示下面还有 k 行数据。

接下来 kk 行，每行两个整数 a，b表示编号为 a 的小格子和编号为 b 的小格子合根了。

格子的编号一行一行，从上到下，从左到右编号。

比如：5×4 的小格子，编号：

1 2 3 4
5 6 7 8
9 10 11 12
13 14 15 16
17 18 19 20

输出描述

输出植物数量。

输入输出样例

示例

5 4
16
2 3
1 5
5 9
4 8
7 8
9 10
10 11
11 12
10 14
12 16
14 18
17 18
15 19
19 20
9 13
13 17

5

其合根情况参考下图：

代码

def find(k):
    while flag[k]:           #不断回溯到父节点
        k = flag[k]
        #print(k)
    return k
    
n,m = map(int,input().split())
k = int(input())
flag = [0] * 10000000
while k:
    x,y = map(int,input().split())
    a = find(x)             
    b = find(y)
    if a < b : flag[b] = a   
    elif a > b : flag[a] = b  #如果大于了表示当前节点与根节点相连，需要赋值
    print("a,b",a,b)
    k -= 1

count = 0
for i in range(1,n*m+1):      #只要把不是节点的都打印出来，不是0的都打印
    if not flag[i]:
        count += 1

print(count)

实现