LeetCode 1722 执行交换操作后的最小汉明距离

给你两个整数数组 source 和 target ，长度都是 n 。还有一个数组 allowedSwaps ，其中每个 allowedSwaps[i] = [ai, bi] 表示你可以交换数组 source 中下标为 ai 和 bi（下标从 0 开始）的两个元素。注意，你可以按 任意 顺序 多次 交换一对特定下标指向的元素。

相同长度的两个数组 source 和 target 间的 汉明距离 是元素不同的下标数量。形式上，其值等于满足 source[i] != target[i] （下标从 0 开始）的下标 i（0 <= i <= n-1）的数量。

在对数组 source 执行 任意 数量的交换操作后，返回 source 和 target 间的 最小汉明距离 。

from typing import *
from collections import defaultdict

class Solution:
    def __init__(self):
        self.arr = None

    def minimumHammingDistance(self, source: List[int],
                               target: List[int], allowedSwaps: List[List[int]]) -> int:
        len1 = len(source)
        self.arr = [i for i in range(len1)]
        for swap in allowedSwaps:
            self.set_union(swap[0], swap[1])
        head2src = defaultdict(lambda: [])
        head2tar = defaultdict(lambda: [])
        head2set = defaultdict(lambda: set())
        for i in range(len1):
            head2src[self.find(i)].append(source[i])
            head2tar[self.find(i)].append(target[i])
            head2set[self.find(i)].add(source[i])
            head2set[self.find(i)].add(target[i])
        res = 0
        for head in head2set:
            res += self.cnt(head2src[head], head2tar[head], head2set[head])
        return res

    def cnt(self, list1, list2, set1):
        res = 0
        dic1 = defaultdict(int)
        dic2 = defaultdict(int)
        for val in list1:
            dic1[val] += 1
        for val in list2:
            dic2[val] += 1
        for val in set1:
            res += abs(dic1[val] - dic2[val])
        return res//2

    def set_union(self, i, j):
        find_j = self.find(j)
        find_i = self.find(i)
        if find_i != find_j:
            self.arr[find_i] = find_j

    def find(self, i):
        if i != self.arr[i]:
            self.arr[i] = self.find(self.arr[i])
        return self.arr[i]