从根到叶的每条路径构成一个数,求这些数的和(Sum Root to Leaf Numbers)

题目描述:

从根到叶子的每条路径构成一个数,求这些数的和。


思路:

直接DFS,每次到叶子加就可以了。


代码实现:

/**
 * 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) {
        int ans=0;
        if(!root) return 0;
        dfsTree(root,root->val,ans);
        return ans;
    }
    
    void dfsTree(TreeNode *fa, int now, int &ans){
        if(!(fa->left||fa->right)) ans+=now;
        else {
            if(fa->left) dfsTree(fa->left,fa->left->val+now*10,ans);
            if(fa->right) dfsTree(fa->right,fa->right->val+now*10,ans);
        }
    }
};

你可能感兴趣的:(LeetCode)