【LeetCode】 617. 合並二叉樹
阿新 • • 發佈:2019-03-07
兩個 val return 將他 子節點 覆蓋 們的 二叉樹 不存在
題目
給定兩個二叉樹,想象當你將它們中的一個覆蓋到另一個上時,兩個二叉樹的一些節點便會重疊。
你需要將他們合並為一個新的二叉樹。合並的規則是如果兩個節點重疊,那麽將他們的值相加作為節點合並後的新值,否則不為 NULL 的節點將直接作為新二叉樹的節點。
示例 1:
輸入: Tree 1 Tree 2 1 2 / \ / \ 3 2 1 3 / \ \ 5 4 7 輸出: 合並後的樹: 3 / 4 5 / \ \ 5 4 7
註意: 合並必須從兩個樹的根節點開始。
思路
使用遞歸算法將兩個二叉樹合並,合並時進行同步遍歷,剛開始可以將t1,t2看做根節點,遞歸開始以後看做子節點,對於相同位置上的一對結點t1和t2,處理策略是:
- 若t1和t2都存在,則只需要保留其中一個結點(如t1),將另一結點的值加到此結點上即可(如t1.val += t2.val)。
- 若t1或t2任一不存在,則合並後的二叉樹對應位置上的結點就是存在的那個了。
- 若n1和n2都不存在,則合並後仍不存在。
代碼
/** * Definition for a binary tree node. * struct TreeNode { * int val; * struct TreeNode *left; * struct TreeNode *right; * }; */ struct TreeNode* mergeTrees(struct TreeNode* t1, struct TreeNode* t2) { if(t1 == NULL) return t2; if(t2 == NULL) return t1; t1->val = t1->val + t2->val; t1->left = mergeTrees(t1->left,t2->left); t1->right = mergeTrees(t1->right,t2->right); return t1; }
【LeetCode】 617. 合並二叉樹