Leetcode 树的路径遍历

题目描述

给定一个仅包含数字0-9的二叉树，每一条从根节点到叶子节点的路径都可以用一个数字表示。

例如根节点到叶子节点的一条路径是1->2->3,那么这条路径就用123来代替。

找出根节点到叶子节点的所有路径表示的数字之和

例如：

    1↵   / ↵  2   3

根节点到叶子节点的路径1->2用数字12代替

根节点到叶子节点的路径1->3用数字13代替

所以答案为12+13=25

 

思路：深度搜索，到达叶子节点，记录一组和即可

class Solution {
public:
    int sumNumbers(TreeNode *root) {
        
        
        if(root==NULL)
            return 0;
        
        int res;
        string temp("");
        DFS(root,res,temp);
        return res;
 
    }
    
    string num2str(int num)
    {
        stringstream ss;
        ss<<num;
        string str;
        ss>>str;
        return str;
    }
    int str2num(string str)
    {
        stringstream ss;
        ss<<str;
        int num;
        ss>>num;
        return num;
    }
    
    
    void DFS(TreeNode * root ,int& res,string temp)
    {
       
        temp+=num2str(root->val);//累加字符串
        if(root->left==NULL && root->right==NULL)  //到达叶子节点
        {
            res+=str2num(temp);//求出该数字
            
        }
        else
        {
            if(root->left!=NULL)
                DFS(root->left,res,temp);
            if(root->right!=NULL)
                DFS(root->right,res,temp);

        }
        
        
    }
    

};