文章目錄

• • 1. 題目
  • 2. 解題

1. 題目

給定一棵有 n 個(gè)結(jié)點(diǎn)的二叉樹(shù)，你的任務(wù)是檢查是否可以通過(guò)去掉樹(shù)上的一條邊將樹(shù)分成兩棵，且這兩棵樹(shù)結(jié)點(diǎn)之和相等。

樣例 1: 輸入: 5/ \10 10/ \2 3 輸出: True 解釋: 5/ 10和: 1510/ \2 3和: 15樣例 2: 輸入: 1/ \2 10/ \2 20 輸出: False 解釋: 無(wú)法通過(guò)移除一條樹(shù)邊將這棵樹(shù)劃分成結(jié)點(diǎn)之和相等的兩棵子樹(shù)。注釋 : 樹(shù)上結(jié)點(diǎn)的權(quán)值范圍 [-100000, 100000]。 1 <= n <= 10000

來(lái)源：力扣（LeetCode）
鏈接：https://leetcode-cn.com/problems/equal-tree-partition
著作權(quán)歸領(lǐng)扣網(wǎng)絡(luò)所有。商業(yè)轉(zhuǎn)載請(qǐng)聯(lián)系官方授權(quán)，非商業(yè)轉(zhuǎn)載請(qǐng)注明出處。

2. 解題

• 自底向上求得每個(gè)節(jié)點(diǎn)的子樹(shù)和，更新于節(jié)點(diǎn)的 val
• 遍歷檢查+剪枝，共計(jì)2次遍歷

class Solution {bool found = false;int allsum; public:bool checkEqualTree(TreeNode* root) {allsum = sum(root);if(allsum&1) return false;//奇數(shù)不可分check(root);return found;}int sum(TreeNode* root){if(!root) return 0;int l = sum(root->left);int r = sum(root->right);return root->val += l+r;}bool check(TreeNode* root){if(!root) return false;if(found) return true;if(root->left && allsum == 2*root->left->val)return found = true;if(root->right && allsum == 2*root->right->val)return found = true;return check(root->left) || check(root->right);} };

40 ms 31.9 MB

---

我的CSDN博客地址 https://michael.blog.csdn.net/

長(zhǎng)按或掃碼關(guān)注我的公眾號(hào)（Michael阿明），一起加油、一起學(xué)習(xí)進(jìn)步！

總結(jié)

以上是生活随笔為你收集整理的LeetCode 663. 均匀树划分（树形DP）的全部?jī)?nèi)容，希望文章能夠幫你解決所遇到的問(wèn)題。

如果覺(jué)得生活随笔網(wǎng)站內(nèi)容還不錯(cuò)，歡迎將生活随笔推薦給好友。