树和二叉树的转换代码python_Python - 将n-ary树转换为二叉树

以下是将n-ary树转换为二叉树的Python代码：

class TreeNode:
    def __init__(self, val):
        self.val = val
        self.left = None
        self.right = None

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

def nary_to_binary(root):
    if not root:
        return None
    node = TreeNode(root.val)
    if len(root.children) > 0:
        node.left = nary_to_binary(root.children[0])
        curr = node.left
        for i in range(1, len(root.children)):
            curr.right = nary_to_binary(root.children[i])
            curr = curr.right
    return node

其中，`TreeNode`表示二叉树节点，`Node`表示n-ary树节点。

我们先根据n-ary树的根节点创建二叉树的根节点，并将其值设为根节点的值。然后，我们判断根节点是否有子节点。如果有，我们将第一个子节点转换为二叉树节点，并将其设置为二叉树节点的左子树。接下来，我们遍历所有的子节点，将它们转换为二叉树节点，并将它们作为当前节点的右子树。最后，我们返回根节点的二叉树表示。