LintCode 二叉树的路径和

1.描述

给定一个二叉树，找出所有路径中各节点相加总和等于给定 目标值 的路径。

一个有效的路径，指的是从根节点到叶节点的路径。

哪家公司问你的这个题？ Airbnb Alibaba Amazon Apple Baidu Bloomberg Cisco Dropbox Ebay Facebook Google Hulu Intel Linkedin Microsoft NetEase Nvidia Oracle Pinterest Snapchat Tencent Twitter Uber Xiaomi Yahoo Yelp Zenefits

感谢您的反馈

样例

给定一个二叉树，和 目标值 = 5:

     1
    / \
   2   4
  / \
 2   3

返回：

[
  [1, 2, 2],
  [1, 4]
]

2.分析

首先找到叶子节点，这样从根节点到叶子结点才是路径，随后判断该路径一路上的节点值

之和是否等于给定目标值。

3.代码

/**
 * Definition of TreeNode:
 * class TreeNode {
 * public:
 *     int val;
 *     TreeNode *left, *right;
 *     TreeNode(int val) {
 *         this->val = val;
 *         this->left = this->right = NULL;
 *     }
 * }
 */
class Solution {
public:
    /**
     * @param root the root of binary tree
     * @param target an integer
     * @return all valid paths
     */
    vector<vector<int>> w;
    void dfs(TreeNode *root,vector<int> num,int sum,int target)
    {
        if(root->left==NULL && root->right==NULL)//若为叶子节点且路径和与目标值相同，路径加入w中，该路径结束
        {
            if(sum==target)
            w.push_back(num);
            return ;
        }
        if(root->left!=NULL)
        {
            int SUM=sum+root->left->val;//此时SUM和为加入左节点值后的值
            vector<int> v(num);
            v.push_back(root->left->val);
            dfs(root->left,v,SUM,target);//左节点值加入后继续遍历左子树
        }
        if(root->right!=NULL)
        {
            int SUM=sum+root->right->val;
            vector<int> v(num);
            v.push_back(root->right->val);
            dfs(root->right,v,SUM,target);
        }
    }
    vector<vector<int>> binaryTreePathSum(TreeNode *root, int target) {
        // Write your code here
        if(root==NULL)
            return w;
        vector<int> A;
        A.push_back(root->val);
        dfs(root,A,root->val,target);
        return w;
    }
};

4.总结

思路是把每条路径都存到小vector中，只有符合目标值的才加入到大vector中。

这里要特别注意红色代码，如果直接调用vector<int> num两处红色代码都是对

num进行的操作，因此需要额外定义一个vector<int> v分别存放左子树和右子树

加入的节点值。