题目:
给你一个字符串 s,以及该字符串中的一些「索引对」数组 pairs,其中 pairs[i] = [a, b] 表示字符串中的两个索引(编号从 0 开始)。
你可以 任意多次交换 在 pairs 中任意一对索引处的字符。
返回在经过若干次交换后,s 可以变成的按字典序最小的字符串。
思路
构建有向图,并对图中的每个联通分量进行内部排序,即可得到最终的结果。
解答:
class Solution:
def bfs(self,res,graph,visited,x):
queue = collections.deque([x])
visited[x] = 1
res.append(x)
while queue:
cur_node = queue.popleft()
for neighbor in graph[cur_node]:
if visited[neighbor]:
continue
visited[neighbor] = 1
res.append(neighbor)
queue.append(neighbor)
def smallestStringWithSwaps(self, s: str, pairs: List[List[int]]) -> str:
# 建有向图
ls=len(s)
graph = [[] for _ in range(ls)]
visited = [0] * ls
for x,y in pairs:
graph[x].append(y)
graph[y].append(x)
res = list(s)
for i in range(ls):
if not visited[i]:
# 获取联通节点
connected_nodes = []
self.bfs(connected_nodes,graph,visited,i)
# 重新赋值
indices = sorted(connected_nodes)
string = sorted(res[node] for node in connected_nodes)
for j,ch in zip(indices,string):
res[j] = ch
return "".join(res)