我想在同一级别的树上显示所有孩子。所以如果我有这样的树:
A
B C D
E F G H I J
例如,等级3将返回E,F,G,H,I和J节点。我在TreeNode
类中有一个方法,它返回给定节点的所有子节点,所以我想做这样的事情:
static Collection<ITreeNode<IProduct>> getOnLevel(ITree<IProduct> tree, int level)
{
Collection<ITreeNode<IProduct>> temp;
int i;
Iterator<ITreeNode<IProduct>> iterator = tree.getRoot().getChildren().iterator();
for(i=0; i<=(level); i++)
{
while(iterator.hasNext())
{
ITreeNode<IProduct> elem = iterator.next();
if(i == (level))
{
temp = elem.getChildren();
return temp;
}
}
}
return tree.getRoot().getChildren();
}
然后我意识到我只是遍历一级孩子,所以我可能不得不以某种方式递归地做到这一点? 在此先感谢,Amar!
答案 0 :(得分:1)
你可以递归地或迭代地做,这取决于你。
我发现递归解决方案稍微容易阅读。它看起来像这样:
static Collection<ITreeNode<IProduct>> getOnLevel(
ITree<IProduct> tree
, int desiredLevel
) {
List<ITreeNode<IProduct>> result = new ArrayList<>();
findOneLevel(tree.getRoot(), desiredLevel, 0, result);
return result;
}
static void findOnLevel(
ITreeNode<IProduct> node
, int desiredLevel
, int currentLevel
, List<ITreeNode<IProduct>> result
) {
if (currentLevel == desiredLevel) {
result.add(node);
return;
}
Iterator<ITreeNode<IProduct>> iterator = node.getChildren().iterator();
while(iterator.hasNext()) {
getOneLevel(iterator.next(), desiredLevel, currentLevel+1, result);
}
}
这种方法非常简单:顶级方法创建一个列表来存储结果,并调用递归findOnLevel
。递归方法检查我们是否已达到所需级别,如果我们这样做,则将当前节点添加到结果中。否则,我们在递归调用中遍历当前节点的所有子节点,并将currentLevel+1
传递给新的当前级别。