leetcode_1490 克隆 N 叉树

468 阅读1分钟

要求

给定一棵 N 叉树的根节点 root ,返回该树的深拷贝(克隆)。

N 叉树的每个节点都包含一个值( int )和子节点的列表( List[Node] )。

class Node {
    public int val;
    public List<Node> children;
}

N 叉树的输入序列用层序遍历表示,每组子节点用 null 分隔(见示例)。

示例 1:

image.png

输入:root = [1,null,3,2,4,null,5,6]
输出:[1,null,3,2,4,null,5,6]

示例 2:

image.png

输入:root = [1,null,2,3,4,5,null,null,6,7,null,8,null,9,10,null,null,11,null,12,null,13,null,null,14]
输出:[1,null,2,3,4,5,null,null,6,7,null,8,null,9,10,null,null,11,null,12,null,13,null,null,14]

提示:

  • 给定的 N 叉树的深度小于或等于 1000。
  • 节点的总个数在 [0, 10^4] 之间

核心代码

"""
# Definition for a Node.
class Node:
    def __init__(self, val=None, children=None):
        self.val = val
        self.children = children if children is not None else []
"""

class Solution:
    def cloneTree(self, root: 'Node') -> 'Node':
        if not root:
            return None
        
        copy = Node(root.val)
        self.dfs(root,copy)
        return copy
    
    def dfs(self,node,copy):
        for node_child in node.children:
            copy_child = Node(node_child.val)
            self.dfs(node_child,copy_child)
            copy.children.append(copy_child)

image.png

解题思路:我们使用深度优先遍历,不断的生成新节点,然后将新节点和旧节点关联,最终克隆出来一个完整的N叉树。