721. 账户合并

最新推荐文章于 2025-05-04 19:41:26 发布

顺在路上

最新推荐文章于 2025-05-04 19:41:26 发布

阅读量216

点赞数

分类专栏：算法文章标签： python leetcode

本文链接：https://blog.youkuaiyun.com/a892445213/article/details/112791924

版权

算法专栏收录该内容

7 篇文章

订阅专栏

这篇博客介绍了如何解决LeetCode的721题——账户合并。作者首先解释了解题思路，包括对原始数据进行整合，使用唯一序号标识邮箱，并对相同名字的邮箱进行合并。然后，通过深度优先搜索（DFS）方法实现代码，最终得到合并后的账户列表，其中每个账户包含名称和有序的邮箱地址。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

给定一个列表accounts，每个元素 accounts[i] 是一个字符串列表，其中第一个元素 accounts[i][0] 是名称 (name)，其余元素是 emails 表示该账户的邮箱地址。

现在，我们想合并这些账户。如果两个账户都有一些共同的邮箱地址，则两个账户必定属于同一个人。请注意，即使两个账户具有相同的名称，它们也可能属于不同的人，因为人们可能具有相同的名称。一个人最初可以拥有任意数量的账户，但其所有账户都具有相同的名称。

合并账户后，按以下格式返回账户：每个账户的第一个元素是名称，其余元素是按顺序排列的邮箱地址。账户本身可以以任意顺序返回。

解题思路

1.先对原始数据进行整合, 对原始所有的邮箱进行编唯一序号, 对所有的邮箱存放名字

2.再次对原始数据进行遍历, 使用同一个名字下的邮箱通过唯一序号进行查并集合并

3.最终会归纳为三个独立的分支, 我们将同一个根的序号放到一起.
此时, 我们通过归纳后的index, 取到邮箱的主人名字, 很容易就得到答案了.

代码

class UnionFinder:
    def __init__(self, n):
        self.father = list(range(n))

    def find(self, x):
        if x != self.father[x]:
            self.father[x] = self.find(self.father[x])
        return self.father[x]


    def merge(self, x, y):
        self.father[self.find(x)] = self.find(y)

def accountsMerge(accounts):

    email_index_dict = {}
    email_name_dict = {}
    for account in accounts:
        name = account[0]
        for email in account[1:]:
            if email not in email_index_dict:
                email_index_dict[email] = len(email_index_dict)
                email_name_dict[email] = name

    uf = UnionFinder(len(email_index_dict))
    for account in accounts:
        first_index = email_index_dict[account[1]]
        for email in account[2:]:
            uf.merge(first_index, email_index_dict[email])

    index_to_email_collection = collections.defaultdict(list)
    for email, index in email_index_dict.items():
        index = uf.find(index)
        index_to_email_collection[index].append(email)

    res = []

    for emails in index_to_email_collection.values():
        res.append([email_name_dict[emails[0]]] + sorted(emails))

    return res

DFS解法

def accountsMerge(accounts):

    def dfs(email, visited, graph, emails):
        if email in visited:
            return
        visited.add(email)
        emails.append(email)
        for neighour in graph[email]:
            dfs(neighour, visited, graph, emails)


    graph = collections.defaultdict(list)
    for account in accounts:
        master = account[1]
        for v in account[2:]:
            graph[master].append(v)
            graph[v].append(master)


    visited = set()
    res = []
    for account in accounts:
        tmp = []
        dfs(account[1], visited, graph, tmp)
        if len(tmp) > 0:
            res.append([account[0]] + sorted(tmp))

    return res