1. 程式人生 > >Leetcode617.Merge Two Binary Trees合併二叉樹

Leetcode617.Merge Two Binary Trees合併二叉樹

給定兩個二叉樹,想象當你將它們中的一個覆蓋到另一個上時,兩個二叉樹的一些節點便會重疊。

你需要將他們合併為一個新的二叉樹。合併的規則是如果兩個節點重疊,那麼將他們的值相加作為節點合併後的新值,否則不為 NULL 的節點將直接作為新二叉樹的節點。

 

struct TreeNode {
     int val;
     TreeNode *left;
     TreeNode *right;
     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
};

class Solution {
public:
	TreeNode * mergeTrees(TreeNode* t1, TreeNode* t2) {
		if (t1 == NULL && t2 == NULL)
			return NULL;
		if (t1 == NULL)
		{
			t2->left = mergeTrees(t1, t2->left);
			t2 ->right = mergeTrees(t1, t2->right);
			return t2;
		}
		else if (t2 == NULL)
		{
			t1 ->left = mergeTrees(t1 ->left, t2);
			t1 ->right = mergeTrees(t1 ->right, t2);
			return t1;
		}
		else
		{
			t1->val = t1->val + t2->val;
			t1 ->left = mergeTrees(t1->left, t2->left);
			t1 ->right = mergeTrees(t1->right, t2->right);
			return t1;
		}
	}
};