题目描述
给定一个列表 accounts,每个元素 accounts[i] 是一个字符串列表,其中第一个元素 accounts[i][0] 是 名称 (name),其余元素是 emails 表示该账户的邮箱地址。
现在,我们想合并这些账户。如果两个账户都有一些共同的邮箱地址,则两个账户必定属于同一个人。请注意,即使两个账户具有相同的名称,它们也可能属于不同的人,因为人们可能具有相同的名称。一个人最初可以拥有任意数量的账户,但其所有账户都具有相同的名称。
合并账户后,按以下格式返回账户:每个账户的第一个元素是名称,其余元素是按顺序排列的邮箱地址。账户本身可以以任意顺序返回。
示例 1:
输入:
accounts = [["John", "johnsmith@mail.com", "john00@mail.com"], ["John", "johnnybravo@mail.com"], ["John", "johnsmith@mail.com", "john_newyork@mail.com"], ["Mary", "mary@mail.com"]]
输出:
[["John", 'john00@mail.com', 'john_newyork@mail.com', 'johnsmith@mail.com'], ["John", "johnnybravo@mail.com"], ["Mary", "mary@mail.com"]]
解释:
第一个和第三个 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']] 也是正确的。
运行结果
解题思路
利用并查集:分别记录email对应的index和name
之后将同一账户下的email进行合并----同一email自然也会再同一类中
按合并后的index将email分类即可
最后将其排序加上name即可
并查集的应用不是很熟练
java代码
class Solution {
public List<List<String>> accountsMerge(List<List<String>> accounts) {
Map<String, Integer> emailToIndex = new HashMap<String, Integer>();
Map<String, String> emailToName = new HashMap<String, String>();
int emailcount=0;
//初始化两个map:分别记录email与index和name的对应关系
for(List<String> account:accounts){
String name=account.get(0);
for(int i=1;i<account.size();i++){
String email=account.get(i);
if(!emailToIndex.containsKey(email)){
emailToIndex.put(email,emailcount++);
emailToName.put(email,name);
}
}
}
//按照email进行index的合并
UnionFind uf=new UnionFind(emailcount);
for(List<String> account:accounts){
String firstemail=account.get(1);
int firstindex=emailToIndex.get(firstemail);
for(int i=2;i<account.size();i++){
String nextemail=account.get(i);
int nextindex=emailToIndex.get(nextemail);
uf.union(firstindex,nextindex);
}
}
//按index将对应的emails分离出来
Map<Integer, List<String>> indexToEmails = new HashMap<Integer, List<String>>();
for(String email:emailToIndex.keySet()){
int index=uf.find(emailToIndex.get(email));
List<String> emails=indexToEmails.getOrDefault(index,new ArrayList<String>());
emails.add(email);
indexToEmails.put(index,emails);
}
//整理输出结果:加上name并对emails进行排序
List<List<String>> res=new ArrayList<List<String>>();
for(List<String> a:indexToEmails.values()){
Collections.sort(a);
List<String> account=new ArrayList<>();
account.add(emailToName.get(a.get(0)));
account.addAll(a);
res.add(account);
}
return res;
}
//并查集类
class UnionFind{
int[] parent;
public UnionFind(int n){
parent=new int[n];
for(int i=0;i<n;i++){
parent[i]=i;
}
}
public void union(int i,int j){
parent[find(j)]=find(i);
}
public int find(int index){
if(parent[index]!=index){
parent[index]=find(parent[index]);
}
return parent[index];
}
}
}
京公网安备 11010502036488号