Leetcode 129 求根到葉子節點數字之和(遞迴)
阿新 • • 發佈:2018-11-11
給定一個二叉樹,它的每個結點都存放一個 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
.
思路:
按照題目要求進行遞迴遍歷,然後找到葉節點的時候將總和加到函式的一個引數上,然後返回這個值。
哈哈,超過了100%的樣例,高興!
程式碼如下:
/** * Definition for a binary tree node. * struct TreeNode { * int val; * TreeNode *left; * TreeNode *right; * TreeNode(int x) : val(x), left(NULL), right(NULL) {} * }; */ class Solution { public: int sumNumbers(TreeNode* root) { if(root==NULL) { return 0; } int sum=0; getSum(root,sum,0); return sum; } void getSum(TreeNode* root,int& sum,int tsum) { if(root->left==NULL&&root->right==NULL) { sum+=tsum*10+root->val; } if(root->left) { getSum(root->left,sum,10*tsum+root->val); } if(root->right) { getSum(root->right,sum,10*tsum+root->val); } } };