叶子结点和分支节点_leetcode No.129 求根到叶子节点数字之和
生活随笔
收集整理的這篇文章主要介紹了
叶子结点和分支节点_leetcode No.129 求根到叶子节点数字之和
小編覺得挺不錯的,現在分享給大家,幫大家做個參考.
題目鏈接:
求根到葉子節點數字之和 - 力扣(LeetCode)?leetcode-cn.com題目描述:
給定一個二叉樹,它的每個結點都存放一個 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.解題思路:
這題初拿到手,思路肯定是找到所有可能的路徑,將路徑和相加就是最終的結果了。
在具體實現的時候,有的人會先把所有路徑的排列都計算出來,然后分別轉換為整數再相加。固然也可以,但是時間、空間復雜度會比較高,我們可以想想有沒有更省事的方法。
觀察一下就知道,其實很多步是可以復用的。就拿示例 2 來說,4->9->5 和 4->9->1里面 4->9 就可以不用重復計算。
具體實現時,在遞歸過程中,到達葉子節點后,count 往回退到上一個值就可以了。
代碼如下:
class Solution { public:int sum = 0;int count = 0;int sumNumbers(TreeNode* root) {dfs(root, count);return sum;}void dfs(TreeNode* root,int count){if(!root){return;}count = count*10 + root->val; // 計算當前路徑下的值if(!root->left && !root->right){ // 到達葉子節點sum += count;count = (count-root->val) / 10; // 往回退到上一個值return;}dfs(root->left, count);dfs(root->right, count);} };如果有任何疑問,歡迎提出。如果有更好的解法,也歡迎告知。
總結
以上是生活随笔為你收集整理的叶子结点和分支节点_leetcode No.129 求根到叶子节点数字之和的全部內容,希望文章能夠幫你解決所遇到的問題。
- 上一篇: python中二进制文件_Python学
- 下一篇: 主题图标_Avada主题网站favico