题目:
给定一个列表 accounts,每个元素 accounts[i] 是一个字符串列表,其中第一个元素 accounts[i][0] 是 名称 (name),其余元素是 emails 表示该帐户的邮箱地址。现在,我们想合并这些帐户。如果两个帐户都有一些共同的邮件地址,则两个帐户必定属于同一个人。请注意,即使两个帐户具有相同的名称,它们也可能属于不同的人,因为人们可能具有相同的名称。一个人最初可以拥有任意数量的帐户,但其所有帐户都具有相同的名称。合并帐户后,按以下格式返回帐户:每个帐户的第一个元素是名称,其余元素是按顺序排列的邮箱地址。accounts 本身可以以任意顺序返回。例子 1:Input:accounts = [["John", "johnsmith@mail.com", "john00@mail.com"], ["John", "johnnybravo@mail.com"], ["John", "johnsmith@mail.com", "john_newyork@mail.com"], ["Mary", "mary@mail.com"]]Output: [["John", 'john00@mail.com', 'john_newyork@mail.com', 'johnsmith@mail.com'], ["John", "johnnybravo@mail.com"], ["Mary", "mary@mail.com"]]Explanation:第一个和第三个 John 是同一个人,因为他们有共同的电子邮件 "johnsmith@mail.com"。第二个 John 和 Mary 是不同的人,因为他们的电子邮件地址没有被其他帐户使用。我们可以以任何顺序返回这些列表,例如答案[['Mary','mary@mail.com'],['John','johnnybravo@mail.com'],['John','john00@mail.com','john_newyork@mail.com','johnsmith@mail.com']]仍然会被接受。来源:力扣(LeetCode)链接:https://leetcode-cn.com/problems/accounts-merge著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
答案:
时间:
20min,未果,逻辑错了。
class UnionFind:def __init__(self,n):self.parent = [i for i in range(n)]self.rank = [n-i for i in range(n)]self.count = ndef find(self,p):while self.parent[p] != self.parent[self.parent[p]]:self.parent[p] = self.parent[self.parent[p]]return self.parent[p]def union(self,p,q):i = self.find(p)j = self.find(q)if i == j:returnif self.rank [i] < self.rank [j]:self.parent[i] = jself.rank [j] += self.rank [i]else:self.parent[j] = iself.rank [i] += self.rank [j]self.count -= 1class Solution:def accountsMerge(self, accounts: List[List[str]]) -> List[List[str]]:n=len(accounts)e2name={}e2id={}Email=UnionFind(n)for i in range(n):name=accounts[i][0]post=accounts[i][1:]for mail in post:e2name[mail]=nameif mail in e2id:Email.union(i,e2id[mail])e2id[mail]= Email.find(i)else:e2id[mail]=itemp=collections.defaultdict(list)for mail,index in e2id.items():father=Email.find(index)temp[father].append(mail)ans=[]for k,v in temp.items():ans.append([accounts[k][0]]+sorted(temp[k]))return ans
要点:
1. 并查集
2. 最后还要找一下爹
3. 最后还要满足题意sort一下
4. 谁作为index
这里只有email是可以hash的,所以所有的dict都用email来做为索引去想就可以了
