for i in [1..9] make_set(i)
set(1, 2): unite(1,2)
set(3): nop
set(4, 5): unite(4,5)
set(3, 2, 6): unite(3,2), unite(2,6)
set(6): nop
set(7, 8): unite(7,8)
set(9, 8): unite(9,8)
Результатом будут множества {1,2,3,6}
, {4,5}
, {7,8,9}
. Получить их можно так: пройдёмся по номерам от 1 до k, для каждого номера x вызывая find_set(x)
— это будет идентификатор соответствующего множества. Его можно использовать, например, с map'ой, которая по id будет отдавать соответствующее множество.
А вообще, в python'е отступами выделяются блоки кода.