知道叶子节点集合怎么生成树

先说一下问题吧
有一颗任意节点的树型结构，级数和节点数不确定，给定叶子节点集合怎么按照原树生成一颗树。要求父节点的子节点如果都不在集合中，则该节点不显示。

我目前想到的算法这样
定义树形结构：
treeNode{
id
name
parentid
set<treeNode>
}

定义一Map<id,treeNode>存放父节点对象。

利用递归遍历叶子节点集合，取父节点id，检查Map中是否存在该节点。
如果存在则取出该父节点，在该节点的叶子节点中添加当前叶子节点，保存在Map中，递归该父节点直至到根节点。
如果不存在，则新建个treeNode,在该节点的叶子节点中添加当前叶子节点，保存在Map中，递归该父节点直至到根节点。
最终取出根节点对象就是最终的树。

不知道大家是否有更优的算法。