w星球的一个种植园，被分成 m * n 个小格子（东西方向m行，南北方向n列）。每个格子里种了一株合根植物。
这种植物有个特点，它的根可能会沿着南北或东西方向伸展，从而与另一个格子的植物合成为一体。
如果我们告诉你哪些小格子间出现了连根现象，你能说出这个园中一共有多少株合根植物吗？
输入格式
第一行，两个整数m，n，用空格分开，表示格子的行数、列数（1<m,n<1000）。
接下来一行，一个整数k，表示下面还有k行数据(0<k<100000)
接下来k行，第行两个整数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
题解
我的代码
s = []
myset = set()
def init_set(N):
global s
s = list(range(N+1))
def find_set(x):
global s
if x != s[x]:
s[x] = find_set(s[x])
return s[x]
def merge_set(x,y):
global s
x = find_set(x)
y = find_set(y)
if x != y:
s[x] = s[y]
m,n = map(int,input().split())
init_set(m*n)
k = int(input())
for _ in range(k):
x,y = map(int,input().split())
merge_set(x,y)
for i in range(1,m*n+1):
x = find_set(s[i])
myset.add(x)
print(len(myset))
另外参考了网友的代码,发现比我简洁好多,其中在并查集的统计上面,不需要利用循环,而是根据计算得出,对我很有启发
import os
import sys
n, m = map(int, input().split())
k = int(input())
fa = list(range(m * n + 10))
def find(x):
if fa[x] != x:
fa[x] = find(fa[x])
return fa[x]
def merge(x, y):
if find(x) != find(y):
fa[find(x)] = find(y)
ans = 0
for i in range(k):
a, b = map(int, input().split())
if find(a) != find(b):
merge(a, b)
ans += 1
print(m * n - ans)
