假设我有一个n元树,如下所示,我需要在每个级别上找到最大值,然后返回[8,7,32]。
8
4 3 7
1 4 3 3 5 6 7 12 32 3 1
我的节点如下所示:公共类节点{
public int val;
public List<Node> children;
public Node() {
}
public Node(int _val,List<Node> _children) {
val=_val;
children=_children;
}
我试着通过递归在每一级得到元素,并找到最大值,但无法做到这一点。
4条答案
按热度按时间z5btuh9x1#
感谢我使用以下解决方案所做的每一个人:
qqrboqgw2#
我们可以通过水平顺序遍历/广度优先搜索得到水平最大值。我们的想法是在一个级别上有一个节点列表/队列。对于此列表中的所有节点,算法执行两个操作:
它计算此级别上的最大值。
它遍历列表/队列的所有节点,获取这些节点的所有子节点,并将它们放入一个新的列表/队列中,然后在下一次迭代中进行处理。
该算法从包含(子树)根的列表/队列开始,在列表/队列为空时结束。
这可以用
Stream
操作:ideone演示
这个实现有两个很好的特性:
它是迭代的,而不是递归的,即算法不受
StackOverflowError
它具有线性时间和内存复杂性通过一点努力,我们甚至可以使算法与泛型一起工作
Node<T extends Comparable<T>>
:ideone演示
pbgvytdp3#
根节点将是其级别的最高节点。对于后续级别,请调用
Collections.sort()
(或将对列表进行排序的任何其他比较)并取最后一个元素(或根据所用排序方法具有最高值的元素)。然后遍历刚刚排序的子节点列表,对于每个节点,对其子节点列表应用相同的处理方法。wh6knrhe4#
递归解决方案非常简单。首先创建一个列表来保存结果。然后遍历所有节点:在每个节点上,将节点的值与同一级别的列表中的值进行比较。如果节点的值更大,则替换列表中的值。