Leetcode:129.求根到葉子節點數字之和
阿新 • • 發佈：2018-11-13
給定一個二叉樹，它的每個結點都存放一個 0-9 的數字，每條從根到葉子節點的路徑都代表一個數字。
例如，從根到葉子節點路徑 1->2->3 代表數字 123。
計算從根到葉子節點生成的所有數字之和。
說明: 葉子節點是指沒有子節點的節點。
示例 1:
輸入: [1,2,3]
1
/ \
2 3
輸出: 25
解釋:
從根到葉子節點路徑 1->2代表數字 12.從根到葉子節點路徑 1->3代表數字 13.因此，數字總和 = 12 + 13 = 25.
示例 2:
輸入:[4,9,0,5,1] 4 / \ 9 0 / \ 5 1 輸出: 1026 解釋: 從根到葉子節點路徑 4->9->5代表數字 495.從根到葉子節點路徑 4->9->1代表數字 491.從根到葉子節點路徑 4->0代表數字 40.因此，數字總和 = 495 + 491 + 40 = 1026.
解題思路：
深度優先搜尋。將所有的葉子結點到根節點的值計算出啦，求和即可。什麼是葉子結點：沒有孩子！
| #define hasLChild(x) (!(x->left==NULL)) #define hasRChild(x) (!(x->right==NULL)) #define hasNoChild(x) (!hasLChild(x)&&!hasRChild(x)) class Solution { public: int sumNumbers(TreeNode* root) { if (root == NULL) return NULL; DFS(root, 0); return sum; } void DFS(TreeNode* root, int val ) { val = val * 10 + root->val; if (hasNoChild(root)) sum += val; if (hasLChild(root)) DFS(root->left, val); if (hasRChild(root)) DFS(root->right, val); } private: int sum = 0; }; |