Python, LintCode, 375. 克隆二叉树

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

class Solution:
    """
    @param root: The root of binary tree
    @return: root of new tree
    """
    def cloneTree(self, root):
        # write your code here
        if root is None:
            return None
        newroot = TreeNode(root.val)
        if root.left:
            newroot.left = self.cloneTree(root.left)
        if root.right:
            newroot.right = self.cloneTree(root.right)
        return root