leetcode 721. 账户合并
问题描述: 给定一个列表 accounts,每个元素 accounts[i] 是一个字符串列表,其中第一个元素 accounts[i][0] 是 名称 (name) ,其余元素是 emails 表示该账户的邮箱地址。
现在,我们想合并这些账户。如果两个账户都有一些共同的邮箱地址,则两个账户必定属于同一个人。请注意,即使两个账户具有相同的名称,它们也可能属于不同的人,因为人们可能具有相同的名称。一个人最初可以拥有任意数量的账户,但其所有账户都具有相同的名称。
合并账户后,按以下格式返回账户:每个账户的第一个元素是名称,其余元素是 按字符 ASCII 顺序排列 的邮箱地址。账户本身可以以 任意顺序 返回。
示例 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']] 也是正确的。
示例 2:
输入: accounts = [["Gabe","Gabe0@m.co","Gabe3@m.co","Gabe1@m.co"],["Kevin","Kevin3@m.co","Kevin5@m.co","Kevin0@m.co"],["Ethan","Ethan5@m.co","Ethan4@m.co","Ethan0@m.co"],["Hanzo","Hanzo3@m.co","Hanzo1@m.co","Hanzo0@m.co"],["Fern","Fern5@m.co","Fern1@m.co","Fern0@m.co"]]
输出: [["Ethan","Ethan0@m.co","Ethan4@m.co","Ethan5@m.co"],["Gabe","Gabe0@m.co","Gabe1@m.co","Gabe3@m.co"],["Hanzo","Hanzo0@m.co","Hanzo1@m.co","Hanzo3@m.co"],["Kevin","Kevin0@m.co","Kevin3@m.co","Kevin5@m.co"],["Fern","Fern0@m.co","Fern1@m.co","Fern5@m.co"]]
思路: 并查集+map
var accountsMerge = function(accounts) {
// 邮箱map:先统计有多少个邮箱 邮箱-新map下标
//名称mapL邮箱对应账户名 邮箱-账户名
const emailToIndex=new Map();
const emailToName=new Map();
let count=0;
for(let account of accounts){
let name=account[0];
for(let i=1;i<account.length;i++){
if(!emailToIndex.has(account[i])){
emailToIndex.set(account[i],count++);
emailToName.set(account[i],name)
}
}
}
// 建立并查集,遍历参数数组,针对在一个数据的邮箱,合并他们在邮箱map中的下标,最终得到邮箱关联的并查集。
let union=new unionSet(count);
for(let account of accounts){
let firstIndex=emailToIndex.get(account[1]);
for(let i=2;i<account.length;i++){
let lastIndex=emailToIndex.get(account[i]);
union.merge(firstIndex,lastIndex)
}
}
//通过上面两个步骤得到的,拼接出并查集下标-邮箱数组的map
const indexToEmails=new Map();
for(let email of emailToIndex.keys()){
let index=union.find(emailToIndex.get(email));
let item=indexToEmails.get(index)?indexToEmails.get(index):[];
item.push(email);
indexToEmails.set(index,item)
};
// 最会遍历上面步骤获得的map,建立新的数组,并返回(别忘了加用户名)
let result=[];
for(let item of indexToEmails.values()){
item.sort();
let res=[];
res.push(emailToName.get(item[0]));
res.push(...item);
result.push(res)
}
return result;
}
class unionSet{
constructor(n){
this.node=new Array(n).fill(0).map((item,index)=>index);
}
find(x){
return this.node[x]=(this.node[x]==x?x:this.find(this.node[x]))
}
merge(x,y){
let fa=this.find(x),fb=this.find(y);
this.node[fa]=fb
}
}