[前端]_一起刷leetcode 721. 账户合并

812 阅读3分钟

大家好,我是挨打的阿木木,爱好算法的前端摸鱼老。最近会频繁给大家分享我刷算法题过程中的思路和心得。如果你也是想提高逼格的摸鱼老,欢迎关注我,一起学习。

题目

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"]]

 

提示:

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

思路

  1. map来记录所有出现过的邮箱和它的索引;
  2. 遍历所有账户的邮箱,判断有没有相同的,如果有的话把两个账户进行链接;
  3. 再次遍历所有账户,取出所有的邮箱,把有关联的放到一起;
  4. 对所有的账户进行去重并排序即可。

实现

/**
 * @param {string[][]} accounts
 * @return {string[][]}
 */
var accountsMerge = function(accounts) {
    const n = accounts.length;
    const uf = new UnionFind(n);

    // 记录每个出现过的元素
    let map = new Map();

    // 建立连接关系
    for (let i = 0; i < n; i++) {
        const list = accounts[i];

        for (let j = 1; j < list.length; j++) {
            // 如果这个账号之前出现过,那么合并了这个账户
            if (map.has(list[j])) {
               uf.merge(i, map.get(list[j]));
            } else {
                map.set(list[j], i);
            }
        }
    }

    map = new Map();
    let result = [];
    // 再次遍历,把相同关系的全部取出来
    for (let i = 0; i < n; i++) {
        const list = accounts[i];
        const index = uf.find(i);

        // 把相同集合的放在一起
        if (map.has(index)) {
            const cur = map.get(index)
            result[cur] = result[cur].concat(list.slice(1));
        } else {
            map.set(index, result.length);
            result.push(list);
        }
    }

    // 第三次遍历,对邮箱进行去重排序
    return result.reduce((total, cur) => {
        const [ name, ...mails  ] = cur;
        total.push([ name ].concat([...new Set(mails)].sort((a, b) => compareString(a, b))));

        return total;
    }, []);
};

// 对比字符串
function compareString(a, b) {
    if (!a || !b) {
        return a ? 1 : -1;
    }

    if (a.charCodeAt() - b.charCodeAt() === 0) {
        return compareString(a.slice(1), b.slice(1));
    }

    return a.charCodeAt() - b.charCodeAt()
}

class UnionFind {
  constructor(n) {
    // 一开始每个元素的父元素都是自己
    this.parent = new Array(n).fill(0).map((item, index) => index);
  }

  // 找到元素的父元素
  find(index) {
    return this.parent[index] = this.parent[index] === index ? index : this.find(this.parent[index]);
  }

  // 把index2的父元素设置为index1的父元素
  merge(index1, index2) {
    this.parent[this.find(index2)] = this.find(index1);
  }
}

看懂了的小伙伴可以点个关注、咱们下道题目见。如无意外以后文章都会以这种形式,有好的建议欢迎评论区留言。