码迷,mamicode.com
首页 > 其他好文 > 详细

617. Merge Two Binary Trees

时间:2017-06-12 10:33:31      阅读:154      评论:0      收藏:0      [点我收藏+]

标签:https   log   get   written   second   ima   process   puts   lan   

https://leetcode.com/problems/merge-two-binary-trees/#/description

 

Given two binary trees and imagine that when you put one of them to cover the other, some nodes of the two trees are overlapped while the others are not.

You need to merge them into a new binary tree. The merge rule is that if two nodes overlap, then sum node values up as the new value of the merged node. Otherwise, the NOT null node will be used as the node of new tree.

Example 1:

Input: 
	Tree 1                     Tree 2                  
          1                         2                             
         / \                       / \                            
        3   2                     1   3                        
       /                           \   \                      
      5                             4   7                  
Output: 
Merged tree:
	     3
	    / 	   4   5
	  / \   \ 
	 5   4   7

 

Note: The merging process must start from the root nodes of both trees.

 

Sol:
 
 
If two trees are not none, then add values from too trees to be the new root. And the left child of the newroot is written by calling the merTrees function recursively. The inputs are left leaf of the first tree and the left leaf of the second tree. Do this the same to the right leaf of newroot.
 
If one of the trees is none while another is not none, return the existing tree. If both trees are none, return none then. 
 
 
# Definition for a binary tree node.
class TreeNode(object):
    def __init__(self, x):
        self.val = x
        self.left = None
        self.right = None

class Solution(object):
    def mergeTrees(self, t1, t2):
        if t1 and t2:
            root = TreeNode(t1.val + t2.val)
            root.left = self.mergeTrees(t1.left, t2.left)
            root.right = self.mergeTrees(t1.right, t2.right)
            return root
        else:
            return t1 or t2

 

 
 
 

617. Merge Two Binary Trees

标签:https   log   get   written   second   ima   process   puts   lan   

原文地址:http://www.cnblogs.com/prmlab/p/6991083.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!