Copy a tree

递归

from collections import deque

class TreeNode(object):
    """docstring for ClassName"""
    def __init__(self, data, left = None, right = None):
        self.data = data
        self.left = left
        self.right = right
    def __str__(self):
        return str(self.data)

def createTree():
    A, B, C, D, E, F, G, H, I = [TreeNode(_) for _ in "ABCDEFGHI"]
    A.left = B 
    A.right = C 
    B.right = D 
    C.left = E 
    C.right = F 
    E.left = G 
    F.left = H 
    F.right = I
    return A
def levelOrder(root):
	q = deque([root])
	result = []
	while q:
		node = q.popleft()
		result.append(node)
		if node.left:
			q.append(node.left)
		if node.right:
			q.append(node.right)
	for i in range(len(result)):
		print(result[i])
	return result	

def copyTree(root):
	if not node:
		return None
	lt = copyTree(root.left)
	rt = copyTree(root.right)
	return TreeNode(root.data, lt, rt)

if __name__ == "__main__"
	root = createTree()
	newTree = copyTree(root)
	levelOrder(newTree)