计算 N 叉树的奇数

Question

我在这个练习中挣扎。 它告诉我计算 N-Ary 中存在的所有奇数。 练习要求我使用递归

看看下面的树

              5                                           
      ________|_____________                        
     |          |           |                     
     20         4           6                     
     |     _____|______                      
     11   |   |  |  |  |                    
          10  2  9  8  7                  
            __|__                        
           |     |                    
           3     1         

里面有12个数字。 在这些数字中，有 6 个是奇数，这是我正在寻找的输出。 但是，我编写的代码并没有解决问题，我不明白为什么。

我实现的函数返回 0，这是不正确的。

def es7(tree):
    n = 0
    for element in tree.f: 
        n = es7(element)
        if element.id % 2 == 1: 
            n += 1 
    return n  
 

class Nodo:
    def __init__(self,V):
        self.id=V
        self.f=[]


################### DA QUI IN GIÙ SONO SOLO FUNZIONI NECESSARIE PER I TEST #####################

def fromLista(lista):
    '''Crea l'albero da una lista [valore, listafigli]
           In cui lista figli contiene alberi o e' la lista vuota. '''
    r=Nodo(lista[0])
    r.f=[fromLista(x) for x in lista[1]]
    return r

def toLista(nodo):
    ''' Converte l'albero in una lista di liste [valore, listafigli]'''
    return [nodo.id, [toLista(x) for x in nodo.f]]   

有什么建议么？

谢谢

Answer 1

解决这个问题的一种简单方法； 就是简单地对N-ary树进行级别顺序遍历。

考虑以下迭代解决方案，它们可以为编写递归解决方案提供一些见解。

要重新编写您的代码，我会在这些行上做一些事情。
考虑下面的这个片段：

def es7(tree):
    q, level = [tree], []
    while any(q):
        level.append([node.id for node in q])
        q = [child for node in q for child in node.f if child]

    ans = 0
    for l in level:
        for num in l:
            if num & 1:
                ans += num
    return ans

考虑下面的代码片段：（非递归解决方案）

from collections import deque

class Node:
    def __init__(self, val=None, children=None):
        self.val = val
        self.children = children

class Solution:
    def levelOrder(self, root: 'Node') -> List[List[int]]:
        if not root:
            return 
        
        q = deque([root])
        level = 0
        result = []
        
        while q:
            result.append([])
            for i in range(len(q)):
                node = q.popleft()
                result[level].append(node.val)
                
                for c in node.children:
                    q.append(c)
            
            level += 1
        
        return result

在上面的代码中， result将根据级别包含n-ary叉树的所有元素。 解析列表，并添加所有奇数。

那将是这样的事情：-

for level in result:
    for num in result:
        if num & 1:
           ans += num

或者，您也可以在遍历树时将所有odd-numbers 。

Answer 2

您在每次迭代时都在遮蔽n 。 你可能想要做的是

def es7(tree):
    n = 0
    for element in tree.f: 
        n += es7(element) # <-- note the `+=`
        if element.id % 2 == 1: 
            n += 1 
    return n

但是，我认为这段代码仍然不完全正确（取决于您对树的表示），您可能会在具有单个奇数的树上实现这一点。