账户合并(Java)

给定一个列表 accounts，每个元素 accounts[i] 是一个字符串列表，其中第一个元素 accounts[i][0] 是 名称 (name)，其余元素是 emails 表示该帐户的邮箱地址。
现在想合并这些帐户。如果两个帐户都有一些共同的邮件地址，则两个帐户必定属于同一个人。但请注意，即使两个帐户具有相同的名称，它们也可能属于不同的人，因为人们可能具有相同的名称。一个人最初可以拥有任意数量的帐户，但其所有帐户都具有相同的名称。合并帐户后，按以下格式返回帐户：每个帐户的第一个元素是名称，其余元素是按顺序排列的邮箱地址。accounts 本身可以以任意顺序返回。

例子 1:
Input:
accounts = [["shen", "loo@mail.com", "shen01@mail.com"], ["shen", "joy@mail.com"], ["shen", "loo@mail.com", "tom@mail.com"], ["Yee", "yee@mail.com"]]
Output: [[shen, loo@mail.com, shen01@mail.com, tom@mail.com],  [Yee, yee@mail.com], [shen, joy@mail.com]]
Explanation:
  第一个和第三个 shen 是同一个人，因为他们有共同的电子邮件 "loo@mail.com"。
  第二个 shen 和 Yee 是不同的人，因为他们的电子邮件地址没有被其他帐户使用。
  可以以任何顺序返回这些列表都会被接受。

注意：
    accounts的长度将在[1，1000]的范围内。
    accounts[i]的长度将在[1，10]的范围内。
    accounts[i][j]的长度将在[1，30]的范围内。

 

package com.loo;

import java.util.Map;
import java.util.HashMap;
import java.util.Set;
import java.util.HashSet;
import java.util.ArrayList;
import java.util.Collections;
import java.util.List;
import java.util.Stack;
import java.util.Arrays;

public class AccountsMerge {
    public static void main(String[] args) {
        String[][] accountsArr = {{"shen", "loo@mail.com", "shen01@mail.com"}, {"shen", "joy@mail.com"}, {"shen", "loo@mail.com", "tom@mail.com"}, {"Yee", "yee@mail.com"}};
        List<List<String>> accounts = new ArrayList<>();
        List<String[]> ls = Arrays.asList(accountsArr);
        for (String[] arr : ls) {
            accounts.add(Arrays.asList(arr));
        }
        List<List<String>> list = accountsMerge(accounts);
        for (List<String> l : list) {
            System.out.println(Arrays.toString(l.toArray()));
        }
    }

    public static List<List<String>> accountsMerge(List<List<String>> accounts) {
        // 先创造一个索引，由邮箱查找用户名
        Map<String , String> mapMailToName =  new HashMap<String , String>();
        // 把每个有关联的邮箱之间连起来，形成一个无向图
        Map<String , ArrayList<String>> mailGraph = new HashMap<String , ArrayList<String>>();
        for (List<String> account : accounts) {
            // 把第一个邮箱把之后的都连起来
            String name = account.get(0);
            String mail = account.get(1);
            for (int i=1;i<account.size();i++) { // 每一个邮箱连到第一个邮箱，第一个邮箱连到每一个邮箱
                mailGraph.computeIfAbsent(account.get(i) , x->new ArrayList<String>()).add(mail);
                mailGraph.computeIfAbsent(mail , x->new ArrayList<String>()).add(account.get(i));
                mapMailToName.put(account.get(i) , name);
            }
        }
        Set<String> mailSet = new HashSet<String>(); // 创建 Set 来存每一个邮箱
        List<List<String>> result = new ArrayList<>();
        for (String email : mailGraph.keySet()) {
            if (!mailSet.contains(email)) { // 如果 Set 里面没有，就进行dfs
                mailSet.add(email);
                Stack<String> sk = new Stack<String>(); // dfs 创建一个栈来维护
                List<String> emailList = new ArrayList<String>();
                sk.push(email);
                while (!sk.isEmpty()) {
                    String emailPop = sk.pop();
                    emailList.add(emailPop);
                    for (String key : mailGraph.get(emailPop)) {
                        if (!mailSet.contains(key)) {
                            sk.push(key);
                            mailSet.add(key);
                        }
                    }
                }
                Collections.sort(emailList);
                emailList.add(0 , mapMailToName.get(email)); // 在0位加入用户名
                result.add(emailList);
            }
        }
        return result;
    }
}