Изменения

for i = 0 to s.size - 1:

s[i] = i//сначала каждый элемент лежит в своем множестве

union(x, y):

if s[x] == s[y]: