[路飞]_leetcode-721-账户合并

[题目地址] [B站地址]

给定一个列表 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"]]

提示：

• 1 <= accounts.length <= 1000
• 2 <= accounts[i].length <= 10
• 1 <= accounts[i][j].length <= 30
• accounts[i][0] 由英文字母组成
• accounts[i][j] (for j > 0) 是有效的邮箱地址

解题思路

本题首先可以将同一个账户下的邮箱放到一个集合中，因为它们肯定是属于当前用户的。
然后在合并当前用户下的邮箱的过程中，同时判断该邮箱是否在其他集合存在，如果在其他集合存在，则说明两个集合属于同一个用户，则对两个集合进行合并。
最后根据获取到的集合，以及邮箱对应的用户名，组装结果数组即可。
那涉及到频繁的集合合并以及获取元素所在的集合的操作，我们可以利用并查集来解题。
如果你对 并查集 还不了解，可以看我这一篇文章 数据结构-并查集,文中介绍了并查集的概念、应用场景以及手写实现的全过程。

动画演示

代码实现

class UnionSet {
  constructor(n){
    // 初始化节点数量数组
    this.size = Array(n).fill(1);
    // 初始化集合列表，每一个节点为一个集合
    this.list = Array(n)
    for(let i = 0;i<n;i++){
      this.list[i] = i;
    }
  }
  // 递归获取元素所在集合根节点
  find(x){
    if(this.list[x]===x) return x;
    // 获取到元素所在集合根节点
    const root = this.find(this.list[x])
    // 将当前节点挂载为根节点子节点，压缩路径
    this.list[x] = root;
    return root;
  }
  // 合并两个元素所在集合
  merge(a,b){
    // 获取两个元素所在集合的根节点
    const rootA = this.find(a),
    rootB = this.find(b);
    // 如果两个根节点相同，则两元素处于同一集合，无需再合并
    if(rootA===rootB) return;
    // 如果 a 所在集合元素数量更多，将 b 所在集合合并到 a 所在集合
    if(this.size[rootA]>this.size[rootB]){
      this.list[rootB] = rootA;
      this.size[rootA] += this.size[rootB]
    }else{
      // 反之将 a 所在集合合并到 b 所在集合
      this.list[rootA] = rootB;
      this.size[rootB] += this.size[rootA]
    }
  }
}
var accountsMerge = function(accounts) {
  // 获取 email => ind 的映射
  const email_ind = new Map(),
  // 获取 email => name 的映射
  email_name = new Map(),
  // 获取账户数量
  accountLen = accounts.length;
  // 初始化邮箱数量为 0
  let count = 0;
  // 遍历账户数组
  for(let i = 0;i<accountLen;i++){
    const item = accounts[i],
    name = item[0];
    // 获取 email => ind，email => name 的映射
    for(let j = 1;j<item.length;j++){
      const email = item[j];
      email_ind.set(email,count++)
      email_name.set(email,name)
    }
  }
  // 实例化并查集
  const unionset = new UnionSet(count);
  // 合并邮箱集合
  for(let i = 0;i<accountLen;i++){
    const item = accounts[i],
    firstInd = email_ind.get(item[1]);
    for(let j = 2;j<item.length;j++){
      unionset.merge(firstInd,email_ind.get(item[j]))
    }
  }
  // 获取合并后的邮箱集合
  const map = new Map();
  email_ind.forEach((ind,email) => {
    const rootInd = unionset.find(ind);
    if(map.has(rootInd)){
      const list = map.get(rootInd);
      list.push(email)
    }else{
      map.set(rootInd,[email])
    }
  })
  // 根据邮箱集合，组装结果数组
  const res = [];
  map.forEach(item => {
    item.sort();
    res.push([email_name.get(item[0]),...item])
  })
  return res;
};

至此我们就完成了 leetcode-721-账户合并

如有任何问题或建议，欢迎留言讨论！