给定一个列表 accounts
,每个元素 accounts[i]
是一个字符串列表,其中第一个元素 accounts[i][0]
是 名称 (name),其余元素是 emails 表示该帐户的邮箱地址。
现在,我们想合并这些帐户。如果两个帐户都有一些共同的邮件地址,则两个帐户必定属于同一个人。请注意,即使两个帐户具有相同的名称,它们也可能属于不同的人,因为人们可能具有相同的名称。一个人最初可以拥有任意数量的帐户,但其所有帐户都具有相同的名称。
合并帐户后,按以下格式返回帐户:每个帐户的第一个元素是名称,其余元素是按顺序排列的邮箱地址。accounts 本身可以以任意顺序返回。
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]
的范围内。
并查集
每个邮箱都是唯一的,所以只要维护邮箱出先在哪一个账户中就可以了,如果邮箱已经出现过,那么就把这两个账户压缩成一个,还要注意一下要求每个人的邮箱按字典序
class Solution {
private:
static const int maxn = 1010;
int f[maxn];
vector<vector<string>> ans; //答案
unordered_map<string, int> m; //存放每个邮箱的位置
unordered_map<int, vector<string>> info;//存放每个人的邮箱
void init(){
for (int i = 0; i < maxn; ++i) f[i] = i;
}
int find(int x){
return f[x] == x ? x : f[x] = find(f[x]);
}
public:
vector<vector<string>> accountsMerge(vector<vector<string>>& act) {
init();
for (int i = 0; i < act.size(); ++i)
for (int j = 1; j < act[i].size(); ++j)
{
//当前邮箱出现过,把当前账户的位置挂在第一次出现位置下
if (m.find(act[i][j]) != m.end())
f[find(i)] = find(m[act[i][j]]);
else
m[act[i][j]] = i;//第一次出现位置i
}
for (auto it : m)
{
int root = find(it.second); //邮箱归属的位置,也就是哪个人
if (info.find(root) == info.end())//第一次出现,添加人名
info[root].push_back(act[root][0]);
info[root].push_back(it.first);
}
for (auto it : info)
{
sort(it.second.begin() + 1, it.second.end());
ans.push_back(it.second);
}
return ans;
}
};