给定一个列表 a c c o u n t s accounts accounts,每个元素 a c c o u n t s [ i ] accounts[i] accounts[i] 是一个字符串列表,其中第一个元素 a c c o u n t s [ i ] [ 0 ] accounts[i][0] accounts[i][0] 是 名称 (name),其余元素是 emails 表示该账户的邮箱地址。
现在,我们想合并这些账户。如果两个账户都有一些共同的邮箱地址,则两个账户必定属于同一个人。请注意,即使两个账户具有相同的名称,它们也可能属于不同的人,因为人们可能具有相同的名称。一个人最初可以拥有任意数量的账户,但其所有账户都具有相同的名称。
合并账户后,按以下格式返回账户:每个账户的第一个元素是名称,其余元素是按顺序排列的邮箱地址。账户本身可以以任意顺序返回。
示例 1:
输入:
accounts = [["John", "[email protected]", "[email protected]"], ["John", "[email protected]"], ["John", "[email protected]", "[email protected]"], ["Mary", "[email protected]"]]
输出:
[["John", '[email protected]', '[email protected]', '[email protected]'], ["John", "[email protected]"], ["Mary", "[email protected]"]]
解释:
第一个和第三个 John 是同一个人,因为他们有共同的邮箱地址 "[email protected]"。
第二个 John 和 Mary 是不同的人,因为他们的邮箱地址没有被其他帐户使用。
可以以任何顺序返回这些列表,例如答案 [['Mary','[email protected]'],['John','[email protected]'],
['John','[email protected]','[email protected]','[email protected]']] 也是正确的。
提示:
a c c o u n t s accounts accounts的长度将在 [ 1 , 1000 ] [1,1000] [1,1000]的范围内。
a c c o u n t s [ i ] accounts[i] accounts[i]的长度将在 [ 1 , 10 ] [1,10] [1,10]的范围内。
a c c o u n t s [ i ] [ j ] accounts[i][j] accounts[i][j]的长度将在 [ 1 , 30 ] [1,30] [1,30]的范围内。
class Solution {
public:
int f[10000];
int find(int x){
return x==f[x]?x:f[x]=find(f[x]);
}
vector<vector<string>> accountsMerge(vector<vector<string>>& accounts) {
int n=accounts.size();
unordered_map<string,int>ma;
for (int i=0;i<10000;i++){
f[i]=i;
}
for (int i=0;i<n;i++){
int m=accounts[i].size();
for (int j=1;j<m;j++){
if (ma.count(accounts[i][j])==0){
ma[accounts[i][j]]=i;
}
f[find(i)]=find(ma[accounts[i][j]]);
}
}
unordered_map<int,vector<string> >ma1;
for (auto &[str,i]:ma){
ma1[find(i)].push_back(str);
}
vector<vector<string>>res;
for (auto &[i,vec]:ma1){
sort(vec.begin(),vec.end());
vector<string>tmp;
tmp.push_back(accounts[find(i)][0]);
tmp.insert(tmp.end(),vec.begin(),vec.end());
res.push_back(tmp);
}
return res;
}
};