[路飞]_js算法:leetcode 721-账户合并

269 阅读2分钟

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"。 
第二个 JohnMary 是不同的人,因为他们的邮箱地址没有被其他帐户使用。
可以以任何顺序返回这些列表,例如答案 [['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
    }
}