python leetcode 1202.交换字符串中的元素

题目描述

解题思路

主要就是考察并查集的思路，并查集使用完后，在单独将，每个集合包含的节点找出来，重新排序，重新赋值即可

代码

class DSU:
    def __init__(self,nodecount):
        self.node_relation_list=[-1]*nodecount#初始化，每个节点的祖先都是自己，记住-1，这里node_count为节点总数
    def findboss(self,node):# 首先，是找到自己所在集合的最上面那一层的祖先，若不为值不为-1，说明当前自己的祖先并不是最终祖先，循环进行上司再去找他的祖先，直到找到最终祖先
        temp=node
        while self.node_relation_list[node]!=-1:
            node=self.node_relation_list[node]
        if temp!=node::#路劲压缩，使得所有节点的祖先都是最终的祖先
            self.node_relation_list[temp]=node
        return node 
    def mergeboss(self,node1,node2)::#查询相互连通的两个人的祖先是不是同一个人
        node1boss=self.findboss(node1)
        node2boss=self.findboss(node2)
        if node1boss!=node2boss::#如果不是，那就合并两个集合，从两人中选举一个新祖先
            self.node_relation_list[node1boss]=node2boss
        return 
class Solution:
    def smallestStringWithSwaps(self, s: str, pairs: List[List[int]]) -> str:
        n=len(s)
        if n<2:
            return s
        dsu=DSU(n)#n个节点数，初始化并查集
        for node1,node2 in pairs:#先用并查集遍历一遍，使得每个节点都找到自己的祖先
            dsu.mergeboss(node1,node2)
        h={
     }
        for i in range(n):#再将所有公共祖先的子节点划分到一起，公共祖先自己也在改集合里
            if dsu.findboss(i) not in h:
                h[dsu.findboss(i)]=[i]
            else:
                h[dsu.findboss(i)].append(i)
        res=list(s)
        for nodes in h.values():
            indices=sorted(nodes)#这里的每个节点都是相互连通的，即可以随意互相置换，直接按题意排序即可
            string=sorted(res[node] for node in nodes)#按最小字典序排列即从小到大
            for index,letter in zip(indices,string):#按排好位置后，放回字母
                res[index]=letter
        return "".join(res)