二叉树中和为某一值的路径

二叉树中和为某一值的路径

题目描述

输入一颗二叉树的跟节点和一个整数，打印出二叉树中结点值的和为输入整数的所有路径。路径定义为从树的根结点开始往下一直到叶结点所经过的结点形成一条路径。(注意: 在返回值的list中，数组长度大的数组靠前)

解题思路

• 用深度优先遍历的思想，path记录符合要求的路径、tmp记录临时路径、node标记当前访问节点、sum记录当前路径和、k记录目标值
• 每遍历到一个节点，就将sum加上该节点的值，并把该节点加入临时路径中
• 如果sum超过了目标值k，不合法的情况，直接返回
• 如果sum==k，要判断node是不是到叶节点了，否则也不合法，返回
• 满足根节点到叶节点的路径和sum==k，才将临时路径加入path
• 如果还未到叶节点，sum也还未到k，则继续往当前节点的左右子节点中遍历
• 递归回退到父节点的时候，tmp中也要弹出相应的元素

class Solution {
public:
    vector<vector<int> > FindPath(TreeNode* root,int expectNumber) {
        vector<vector<int>>path;
        vector<int>tmp;
        if(root==NULL) return path;
        dfs(path,tmp,root,0,expectNumber);
        return path;
    }
    void dfs(vector<vector<int>>&path,vector<int>tmp,TreeNode *node,int sum,int k){
        if(node==NULL) return ;
        sum = sum + node->val;
        tmp.push_back(node->val);
        if(sum>k){    //超过了
            return ;
        }else if(sum==k){
            //还没到叶节点
            if(node->left!=NULL || node->right!=NULL) return ;
            path.push_back(tmp);
            return ;
        }
        dfs(path,tmp,node->left,sum,k);
        dfs(path,tmp,node->right,sum,k);
        tmp.pop_back();//回溯
        return ;
    }
};