大家好,我是挨打的阿木木,爱好算法的前端摸鱼老。最近会频繁给大家分享我刷算法题过程中的思路和心得。如果你也是想提高逼格的摸鱼老,欢迎关注我,一起学习。
题目
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"]]
提示:
1 <= accounts.length <= 10002 <= accounts[i].length <= 101 <= accounts[i][j].length <= 30accounts[i][0]由英文字母组成accounts[i][j] (for j > 0)是有效的邮箱地址
思路
- 用
map来记录所有出现过的邮箱和它的索引; - 遍历所有账户的邮箱,判断有没有相同的,如果有的话把两个账户进行链接;
- 再次遍历所有账户,取出所有的邮箱,把有关联的放到一起;
- 对所有的账户进行去重并排序即可。
实现
/**
* @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);
}
}
看懂了的小伙伴可以点个关注、咱们下道题目见。如无意外以后文章都会以这种形式,有好的建议欢迎评论区留言。