账户合并

给定一个列表 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] [11000]的范围内。
a c c o u n t s [ i ] accounts[i] accounts[i]的长度将在 [ 1 , 10 ] [1,10] [110]的范围内。
a c c o u n t s [ i ] [ j ] accounts[i][j] accounts[i][j]的长度将在 [ 1 , 30 ] [1,30] [130]的范围内。

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;
    }
};

猜你喜欢

转载自blog.csdn.net/weixin_43601103/article/details/112763190