求根到葉子節點數字之和

 

給定一個二叉樹，它的每個結點都存放一個 0-9 的數字，每條從根到葉子節點的路徑都代表一個數字。

例如，從根到葉子節點路徑 1->2->3 代表數字 123。

計算從根到葉子節點生成的所有數字之和。

說明: 葉子節點是指沒有子節點的節點。

示例 1:

輸入: [1,2,3]
    1
   / \
  2   3
輸出: 25

---

還蠻簡單的，就是遞歸調用

/**
 * 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;
        return sum(root,0);
        
    }
    
    int sum(TreeNode *a,int value){
        int parm=value*10+a->val;
        if(a->left!=NULL && a->right==NULL)
            return sum(a->left,parm);
        else if(a->left==NULL && a->right!=NULL)
            return sum(a->right,parm);
        else if(a->left!=NULL && a->right!=NULL)
            return sum(a->right,parm)+sum(a->left,parm);
        else
            return parm;

    }
};