python交换字符串里的元素_力扣 leetcode 1202. 交换字符串中的元素 （python）
Topic:
给你一个字符串 s，以及该字符串中的一些「索引对」数组 pairs，其中 pairs[i] = [a, b] 表示字符串中的两个索引(编号从 0 开始)。
你可以 任意多次交换 在 pairs 中任意一对索引处的字符。
返回在经过若干次交换后，s 可以变成的按字典序最小的字符串。
Example_1:
输入：s = “dcab”, pairs = [[0,3],[1,2]]
输出：“bacd”
解释：
交换 s[0] 和 s[3], s = “bcad”
交换 s[1] 和 s[2], s = “bacd”
Example_2:
输入：s = “dcab”, pairs = [[0,3],[1,2],[0,2]]
输出：“abcd”
解释：
交换 s[0] 和 s[3], s = “bcad”
交换 s[0] 和 s[2], s = “acbd”
交换 s[1] 和 s[2], s = “abcd”
Example_3:
输入：s = “cba”, pairs = [[0,1],[1,2]]
输出：“abc”
解释：
交换 s[0] 和 s[1], s = “bca”
交换 s[1] 和 s[2], s = “bac”
交换 s[0] 和 s[1], s = “abc”
Solution:
本题整体思路基于并查集模板实现
首先是建图的过程：
将s中的值按照索引模拟成图的n个节点
每个索引是一个节点
之后通过pairs中的数组
将对应的可以互换的节点连通起来
之后是寻找联通的所有节点用于保存连通图
将根节点相同的放在一起
以根节点作为键
值为数组，存储以该根节点为根的并查集中所有节点下标
最后分别对每个根节点对应的连通关系按照字典序进行单独排列
将排序好的结果整合到res中即可完成
Code：
class UnionFind:
def __init__(self,s):
# 创建并查集图
self.father = {i:i for i in range(len(s))}
def find(self,x):
# 查找根节点
root = x
while self.father[root] != root:
root = self.father[root]
# 路径压缩
while x != root:
original_father = self.father[x]
self.father[x] = root
x = original_father
return root
def merge(self, x, y):
# 合并节点
root_x, root_y = self.find(x), self.find(y)
if root_x != root_y:
self.father[root_x] = root_y
class Solution:
def smallestStringWithSwaps(self, s: str, pairs: List[List[int]]) -> str:
# 建图
uf = UnionFind(s)
for x, y in pairs:
uf.merge(x, y)
# 获取联通节点
connected = collections.defaultdict(list)
for node in range(len(s)):
connected[uf.find(node)].append(node)
# 重新赋值
res = list(s)
for nodes in connected.values():
last = nodes
string = sorted(res[d] for d in nodes)
for e, f in zip(last,string):
res[e] = f
return "".join(res)
Result：
