计算给定树中权重为 2 的幂的节点

在计算树中权重为 2 的幂的节点时，我们需要遍历整棵树，找到所有权重为 2 的幂的节点。一般情况下，我们可以使用递归函数来完成树的遍历，同时使用位运算来判断某个数是否是 2 的幂。

以下是计算给定树中权重为 2 的幂的节点的 Python 代码片段:

def find_power_of_two_nodes(root):
    """
    给定一个树的根节点，返回权重为 2 的幂的节点
    """
    power_of_two_nodes = []
    
    def dfs(node, weight):
        """
        递归函数，遍历树中的节点
        """
        if not node:
            return
        
        # 判断节点是否是权重为 2 的幂
        if weight & (weight - 1) == 0 and weight != 0:
            power_of_two_nodes.append(node.val)
        
        # 遍历左右子树
        dfs(node.left, weight * 2)
        dfs(node.right, weight * 2)
    
    dfs(root, 1)
    
    return power_of_two_nodes

以上代码中的 find_power_of_two_nodes 函数接受一个树的根节点作为参数，并返回一个包含所有权重为 2 的幂的节点值的列表。该函数使用了嵌套的 dfs 函数来遍历整棵树。dfs 函数需要传入当前节点和该节点的权重。在递归函数中，我们通过位运算判断该节点的权重是否是 2 的幂。如果是，则将该节点的值添加到 power_of_two_nodes 列表中，并继续遍历左右子树。

Markdown 格式如下：

计算给定树中权重为 2 的幂的节点

在计算树中权重为 2 的幂的节点时，我们需要遍历整棵树，找到所有权重为 2 的幂的节点。一般情况下，我们可以使用递归函数来完成树的遍历，同时使用位运算来判断某个数是否是 2 的幂。

以下是计算给定树中权重为 2 的幂的节点的 Python 代码片段:

def find_power_of_two_nodes(root):
    """
    给定一个树的根节点，返回权重为 2 的幂的节点
    """
    power_of_two_nodes = []
    
    def dfs(node, weight):
        """
        递归函数，遍历树中的节点
        """
        if not node:
            return
        
        # 判断节点是否是权重为 2 的幂
        if weight & (weight - 1) == 0 and weight != 0:
            power_of_two_nodes.append(node.val)
        
        # 遍历左右子树
        dfs(node.left, weight * 2)
        dfs(node.right, weight * 2)
    
    dfs(root, 1)
    
    return power_of_two_nodes

以上代码中的 find_power_of_two_nodes 函数接受一个树的根节点作为参数，并返回一个包含所有权重为 2 的幂的节点值的列表。该函数使用了嵌套的 dfs 函数来遍历整棵树。dfs 函数需要传入当前节点和该节点的权重。在递归函数中，我们通过位运算判断该节点的权重是否是 2 的幂。如果是，则将该节点的值添加到 power_of_two_nodes 列表中，并继续遍历左右子树。