leetcode721. 账户合并
给定一个列表
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']] 也是正确的。
/**
* @param {string[][]} accounts
* @return {string[][]}
*/
var accountsMerge = function(accounts) {
const emailToIndex = new Map();//邮箱+坐标
const emailToName = new Map();//邮箱+名称
let emailsCount = 0;
for (const account of accounts) {//拿到里面的二维数组
const name = account[0]; //拿到每一个集合的名称
const size = account.length;//
for (let i = 1; i < size; i++) {//从第一个开始拿到
const email = account[i];//拿到邮箱
if (!emailToIndex.has(email)) {//开始判断如果emial没有存进去
emailToIndex.set(email, emailsCount++);//坐标存进去++
emailToName.set(email, name);//存名称
}
}
}
// 至此我们就把 邮箱坐标,邮箱+名称这两个map表格创建完了
// 两个map创建完了,拿到邮箱数量了,这样我们就可以创建并查集了
const uf = new UnionFind(emailsCount);//调用并查集
// 根据两个map对邮箱进行合并
for (const account of accounts) {//
const firstEmail = account[1];// 拿到第一个邮箱
const firstIndex = emailToIndex.get(firstEmail);//拿到他在map里面的下标
const size = account.length;//
for (let i = 2; i < size; i++) {// 开始拿第二个邮箱,开始跟第一个邮箱进行连通
const nextEmail = account[i];//拿到后面的每一个邮箱
const nextIndex = emailToIndex.get(nextEmail);//根据这个邮箱,拿到他对应的在map里面的下标
uf.union(firstIndex, nextIndex);//进行连通
}
}
// 全给拿出来
const indexToEmails = new Map();//K是并查集祖先坐标 v是邮箱集合,如果坐标当作k,那么我们如何拿到k
for (const email of emailToIndex.keys()) {//从emailToIndex拿坐标,所有的邮箱就拿到了,拿到每一个邮箱
const index = uf.find(emailToIndex.get(email));//拿到祖先的坐标
const account = indexToEmails.get(index) ? indexToEmails.get(index) : [];//拿坐标
account.push(email);//把邮箱添加完
indexToEmails.set(index, account);//根据祖先节点分好组
}
//做用户之间的合并
const merged = [];//建一个数组
for (const emails of indexToEmails.values()) { //获取到邮箱的集合,合并之前要按着ascll进行排序
emails.sort();
const name = emailToName.get(emails[0]);// 拿到名称
const account = [];//
account.push(name);// 添加名字
account.push(...emails);// 添加邮箱
merged.push(account);//把数组集合添加进去
}
return merged;//完成用户之间的合并
};
class UnionFind {
constructor (n) {
this.parent = new Array(n).fill(0).map((element, index) => index);
}
union (index1, index2) {
this.parent[this.find(index2)] = this.find(index1);
}
find (index) {
if (this.parent[index] !== index) {
this.parent[index] = this.find(this.parent[index]);
}
return this.parent[index];
}
}
代码解读:
使用了并查集(Union-Find)数据结构。在这段代码中,并查集用于合并具有相同电子邮件地址的不同账户。具体来说,它通过以下步骤实现:
创建一个映射表(Map)
emailToIndex
,将每个电子邮件地址映射到一个唯一的索引值。同时,创建另一个映射表emailToName
,将每个电子邮件地址映射到对应的账户名称。遍历所有账户,将每个账户中的电子邮件地址添加到
emailToIndex
和emailToName
映射表中。创建一个并查集实例
uf
,其大小等于电子邮件地址的数量。再次遍历所有账户,对于每个账户中的电子邮件地址,找到其在
emailToIndex
映射表中的索引值,并将这些索引值在并查集中进行合并。最后,遍历
emailToIndex
映射表,根据并查集中的祖先节点将电子邮件地址分组,并将每组电子邮件地址按字母顺序排序。同时,从emailToName
映射表中获取每个分组对应的账户名称。将分组后的电子邮件地址和账户名称组合成一个新的数组,并将其添加到结果数组
merged
中。返回结果数组
merged
,其中包含了合并后的用户账户信息。