Leetcode 721. 账户合并

1.题目基本信息

1.1.题目描述

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

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

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

1.2.题目地址

https://leetcode.cn/problems/accounts-merge/description/

2.解题方法

2.1.解题思路

并查集+哈希表

2.2.解题步骤

第一步,构建email->username的映射,同时构建邮箱之间的并查集(同一个账户的邮箱属于一个集合)

第二步,根据并查集的映射,构建rootEmail->emails的映射

第三步,根据rootEmail->emails的映射构建题解并返回

3.解题代码

Python代码

python 复制代码
# # ==> 并查集模板(附优化)
class UnionFind():
    def __init__(self):
        self.roots={}
        self.setCnt=0   # 连通分量的个数
        # Union优化:存储根节点主导的集合的总节点数
        self.rootSizes={}
    
    def add(self,x):
        if x not in self.roots:
            self.roots[x]=x
            self.rootSizes[x]=1
            self.setCnt+=1
    
    def find(self,x):
        root=x
        while root != self.roots[root]:
            root=self.roots[root]
        # 优化:压缩路径
        while x!=root:
            temp=self.roots[x]
            self.roots[x]=root
            x=temp
        return root
    
    def union(self,x,y):
        rootx,rooty=self.find(x),self.find(y)
        if rootx!=rooty:
            # 优化:小树合并到大树上
            if self.rootSizes[rootx]<self.rootSizes[rooty]:
                self.roots[rootx]=rooty
                self.rootSizes[rooty]+=self.rootSizes[rootx]
            else:
                self.roots[rooty]=rootx
                self.rootSizes[rootx]+=self.rootSizes[rooty]
            self.setCnt-=1

from collections import defaultdict

class Solution:
    def accountsMerge(self, accounts: List[List[str]]) -> List[List[str]]:
        uf=UnionFind()
        # 第一步,构建email->username的映射,同时构建邮箱之间的并查集(同一个账户的邮箱属于一个集合)
        emailToUserMap={}
        for account in accounts:
            username=account[0]
            firstEmail=account[1]
            for email in account[1:]:
                emailToUserMap[email]=username
                uf.add(email)
                uf.union(firstEmail,email)        
        # 第二步,根据并查集的映射,构建rootEmail->emails的映射
        rootToEmailsMap=defaultdict(list)
        for email in uf.roots.keys():
            rootEmail=uf.find(email)
            rootToEmailsMap[rootEmail].append(email)
        # print(rootToEmailsMap)
        # 第三步,根据rootEmail->emails的映射构建题解并返回
        result=[]
        for key,emails in rootToEmailsMap.items():
            result.append([emailToUserMap[key]]+sorted(emails))
        # print(result)
        return result

4.执行结果

相关推荐
trayvontang43 分钟前
Python虚拟环境与包管理工具(uv、Conda)
python·conda·uv·虚拟环境·miniconda·miniforge
伊织code1 小时前
pdfminer.six
python·pdf·图片·提取·文本·pdfminer·pdfminer.six
楼田莉子1 小时前
C++算法题目分享:二叉搜索树相关的习题
数据结构·c++·学习·算法·leetcode·面试
hqxstudying1 小时前
JAVA项目中邮件发送功能
java·开发语言·python·邮件
Q_Q5110082852 小时前
python的软件工程与项目管理课程组学习系统
spring boot·python·django·flask·node.js·php·软件工程
pusue_the_sun2 小时前
数据结构——栈和队列oj练习
c语言·数据结构·算法··队列
大锦终2 小时前
【算法】模拟专题
c++·算法
合作小小程序员小小店2 小时前
SDN安全开发环境中常见的框架,工具,第三方库,mininet常见指令介绍
python·安全·生成对抗网络·网络安全·网络攻击模型
后台开发者Ethan2 小时前
Python需要了解的一些知识
开发语言·人工智能·python
Xの哲學2 小时前
Perf使用详解
linux·网络·网络协议·算法·架构