java - 将具有多个值的 map 转换为树？

Question

给定一组随机分布的键，每个键都映射到一组值，您将如何将其转换为多棵树？

示例数据集

• NB₂ => {NC₂ ND₂
  
• ND₁ => {NG₁ NH₁}
  
• NA₁ => {NB₁}
• NB₁ => {NC₁ ND₁ NE₁}
  
• NA₂ => {NB₂}
  
• NC₁ => {NF₁}
  
• NE₁ => {NI₁ NJ₁ NK1}
  

NA₁的结果树

NA1
`-- NB1
    |-- NC1
    |   `-- NF1
    |-- ND1
    |   |-- NG1
    |   `-- NH1
    `-- NE1
        |-- NI1
        |-- NJ1
        `-- NK1

NA₂的结果树

NA2
`-- NB2
    |-- NC2
    `-- ND2

Answer 1

我不知道有任何库方法可以进行这种转换。这就是我的做法。 IMO，这非常简单。

public class Tree {
    public Tree(String key) {
        // ...
    }
    public void addChild(Tree child) {
        // ...
    }
}

public Set<Tree> transform(Map<String, List<String>> input) {
    // Potential tree roots.  We start with all LHS keys as potential roots,
    // and eliminate them when we see their keys on the RHS.
    Set<String> roots = new HashSet<String>(input.keySet());

    // This map associates keys with the tree nodes that we create for them
    Map<String, Tree> map = new HashMap<String, Tree>();

    for (Map.Entry<String, List<String>> entry : input.entrySet()) {
        String key = entry.getKey();
        List<String> childKeys = entry.getValue();
        Tree tree = map.get(key);
        if (tree == null) {
            tree = new Tree(key);
            map.put(key, tree);
        }
        for (String childKey : childKeys) {
            roots.remove(childKey);
            Tree child = map.get(childKey);
            if (child == null) {
                child = new Tree(childKey);
                map.put(childKey, child);
            }
            tree.addChild(child);
        }
    }
    Set<Tree> res = new HashSet<Tree>(roots.size());
    for (String key : roots) {
        res.add(map.get(key));
    }
    return res;
}

编辑:请注意，如果输入表示一组 DAG(有向无环图)，则此算法将“有效”。但是，我刚刚意识到生成的一组树将共享输入数据中任何公共(public)子树的 TreeNode 实例。

请注意我还没有调试这段代码:-)