给定一个列表 accounts
,每个元素 accounts[i]
是一个字符串列表,其中第一个元素 accounts[i][0]
是 名称 (name),其余元素是 emails 表示该帐户的邮箱地址。
现在,我们想合并这些帐户。如果两个帐户都有一些共同的邮件地址,则两个帐户必定属于同一个人。请注意,即使两个帐户具有相同的名称,它们也可能属于不同的人,因为人们可能具有相同的名称。一个人最初可以拥有任意数量的帐户,但其所有帐户都具有相同的名称。
合并帐户后,按以下格式返回帐户:每个帐户的第一个元素是名称,其余元素是按顺序排列的邮箱地址。accounts 本身可以以任意顺序返回。
例子 1:
Input:
accounts = [["John", "johnsmith@mail.com", "john00@mail.com"], ["John", "johnnybravo@mail.com"], ["John", "johnsmith@mail.com", "john_newyork@mail.com"], ["Mary", "mary@mail.com"]]
Output: [["John", 'john00@mail.com', 'john_newyork@mail.com', 'johnsmith@mail.com'], ["John", "johnnybravo@mail.com"], ["Mary", "mary@mail.com"]]
Explanation:
第一个和第三个 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']]仍然会被接受。
注意:
accounts
的长度将在[1,1000]
的范围内。accounts[i]
的长度将在[1,10]
的范围内。accounts[i][j]
的长度将在[1,30]
的范围内。思路:这道题需要用Union Find的方法来做,关键是要把能连接起来的emails连接到一起,为了把这些emails组织到一起,需要有对应的parent(其实就是祖先结点),下面以一个举例说明:
a b c // b, c的父亲是a
d e f // e, f的父亲是d
g a d // 现在abc, def全部合并到g
当解析完 a b c 后的节点序列:a->b表示a的父亲是b
a->a
b->a
c->a
当解析完 d e f 后的节点序列
d->d
e->d
f->d
当解析完 g a d 后的节点序列
g->g
a->g
d->g
参考代码:
string find(string p, unordered_map<string, string> &parents) {
return parents[p] == p ? p : (find(parents[p], parents));
}
vector<vector<string>> accountsMerge(vector<vector<string>>& acts) {
unordered_map<string, string> owner;
unordered_map<string, string> parents;
unordered_map<string, set<string>> unions;
//initial parents and owner
for (int i = 0; i < acts.size(); i++) {
for (int j = 1; j < acts[i].size(); j++) {
parents[acts[i][j]] = acts[i][j];
owner[acts[i][j]] = acts[i][0];
}
}
//update parents
for (int i = 0; i < acts.size(); i++) {
string p = find(acts[i][1], parents);
for (int j = 2; j < acts[i].size(); j++) {
parents[find(acts[i][j],parents)] = p;
}
}
// calculate union operation
for (int i = 0; i < acts.size(); i++) {
for (int j = 1; j < acts[i].size(); j++) {
unions[find(acts[i][j], parents)].insert(acts[i][j]);
}
}
vector<vector<string>> res;
for (auto &p : unions) {
vector<string> emails(p.second.begin(), p.second.end());
emails.insert(emails.begin(), owner[p.first]);
res.push_back(emails);
}
return res;
}