题目:
给定一个列表 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 = n
def 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:
return
if self.rank [i] < self.rank [j]:
self.parent[i] = j
self.rank [j] += self.rank [i]
else:
self.parent[j] = i
self.rank [i] += self.rank [j]
self.count -= 1
class 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]=name
if mail in e2id:
Email.union(i,e2id[mail])
e2id[mail]= Email.find(i)
else:
e2id[mail]=i
temp=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来做为索引去想就可以了