并查集(字符串形式)

链接

class Solution {
    // 使用 Map 来保存每个节点的父节点
    Map<String, String> par = new HashMap<>();

    public String[] trulyMostPopular(String[] names, String[] synonyms) {
        // 初始化每个节点
        String[] nameArray = Arrays.stream(names)
        .map(ele -> ele.substring(0, ele.indexOf("("))).toArray(String[]::new);
        
        // 初始化 
        init(nameArray);

        // 合并
        for (String synonym: synonyms) {
            String[] edge = synonym.replace("(", " ").replace(")", " ").trim().split(",");
            union(edge[0], edge[1]);
        }

        // 计算数字
        Map<String, Integer> count = new HashMap<>();
        for (String item: names) {
            int leftIdx = item.indexOf("(");
            String name = item.substring(0, leftIdx);
            int num = Integer.parseInt(item.substring(leftIdx + 1, item.length() - 1));
            name = find(name);
            count.put(name, count.getOrDefault(name, 0) + num);
        }
        return count.entrySet().stream().map(ele -> String.format("%s(%d)", ele.getKey(), ele.getValue())).toArray(String[]::new);
    }

    // 初始化
    public void init(String[] names) {
        for (String name: names) {
            par.put(name, name);
        }
    }

    // 查看
    public String find(String x) {
        if (x == null) return null;
        if (x.equals(par.get(x))) return x;
        par.put(x, find(par.get(x)));
        return par.get(x);
    }

    // 合并
    public void union(String x, String y) {
        String parx = find(x); 
        String pary = find(y);
        // 注意 有的name 可能没有进入init 这里注意判空
        if (parx == null) parx = x; 
        if (pary == null) pary = y;
        if (pary.equals(parx)) return ;
        else {
            if (parx.compareTo(pary) > 0) {   // x的字段序 大于 y的字典序
                par.put(parx, pary);
            } else {
                par.put(pary, parx);
            }
        }
    }
}