题目描述

给定一个列表 accounts,每个元素 accounts[i] 是一个字符串列表,其中第一个元素 accounts[i][0] 是 名称 (name),其余元素是 emails 表示该账户的邮箱地址。

现在,我们想合并这些账户。如果两个账户都有一些共同的邮箱地址,则两个账户必定属于同一个人。请注意,即使两个账户具有相同的名称,它们也可能属于不同的人,因为人们可能具有相同的名称。一个人最初可以拥有任意数量的账户,但其所有账户都具有相同的名称。

合并账户后,按以下格式返回账户:每个账户的第一个元素是名称,其余元素是按顺序排列的邮箱地址。账户本身可以以任意顺序返回。

示例 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']] 也是正确的。
运行结果

解题思路
利用并查集:分别记录email对应的index和name
之后将同一账户下的email进行合并----同一email自然也会再同一类中
按合并后的index将email分类即可
最后将其排序加上name即可
并查集的应用不是很熟练
java代码

class Solution {
    public List<List<String>> accountsMerge(List<List<String>> accounts) {
        Map<String, Integer> emailToIndex = new HashMap<String, Integer>();
        Map<String, String> emailToName = new HashMap<String, String>();
        int emailcount=0;
        //初始化两个map:分别记录email与index和name的对应关系
        for(List<String> account:accounts){
            String name=account.get(0);
            for(int i=1;i<account.size();i++){
                String email=account.get(i);
                if(!emailToIndex.containsKey(email)){
                    emailToIndex.put(email,emailcount++);
                    emailToName.put(email,name);
                }
            }
        }
        //按照email进行index的合并
        UnionFind uf=new UnionFind(emailcount);
        for(List<String> account:accounts){
            String firstemail=account.get(1);
            int firstindex=emailToIndex.get(firstemail);
            for(int i=2;i<account.size();i++){
                String nextemail=account.get(i);
                int nextindex=emailToIndex.get(nextemail);
                uf.union(firstindex,nextindex);
            }
        }
        //按index将对应的emails分离出来
        Map<Integer, List<String>> indexToEmails = new HashMap<Integer, List<String>>();
        for(String email:emailToIndex.keySet()){
            int index=uf.find(emailToIndex.get(email));
            List<String> emails=indexToEmails.getOrDefault(index,new ArrayList<String>());
            emails.add(email);
            indexToEmails.put(index,emails);
        }
        //整理输出结果:加上name并对emails进行排序
        List<List<String>> res=new ArrayList<List<String>>();
        for(List<String> a:indexToEmails.values()){
            Collections.sort(a);
            List<String> account=new ArrayList<>();
            account.add(emailToName.get(a.get(0)));
            account.addAll(a);
            res.add(account);
        }
        return res;
    }
    //并查集类
    class UnionFind{
        int[] parent;
        public UnionFind(int n){
            parent=new int[n];
            for(int i=0;i<n;i++){
                parent[i]=i;
            }
        }
        public void union(int i,int j){
            parent[find(j)]=find(i);
        }
        public int find(int index){
            if(parent[index]!=index){
                parent[index]=find(parent[index]);
            }
            return parent[index];
        }
    }

}