首页 > 代码库 > LeetCode:Path Sum II

LeetCode:Path Sum II

      Given a binary tree and a sum, find all root-to-leaf paths where each path‘s


sum equals the given sum.


For example:


Given the below binary tree andsum = 22,

              5
             /             4   8
           /   /           11  13  4
         /  \    /         7    2  5   1

return

[
   [5,4,11,2],
   [5,8,4,5]
]解题思路:    通过遍历树保存从根到叶子节点的路径和路径和,然后判断其和是否等于sum即可.解题代码:
/**
 * Definition for binary tree
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
 * };
 */
class Solution {
public:
    void dfs(TreeNode *rt,long long sum,vector<vector<int> > &ans,vector<int> &path)
    {
        path.push_back(rt->val);
        if (rt->left == rt->right && rt->left == NULL)
        {
            if (sum == rt->val)
                ans.push_back(vector<int>(path.begin(),path.end()));
            path.erase(path.end()-1);
            return ;
        }
        if (rt->left)
            dfs(rt->left,sum - rt->val,ans,path);
        if (rt->right)
            dfs(rt->right,sum - rt->val,ans,path);
        path.erase(path.end()-1);
    }
    vector<vector<int> > pathSum(TreeNode *root, int sum) 
    {
        vector<vector<int> > ans ;
        if (!root)
            return ans ;
        vector<int> path;
        dfs(root,sum,ans,path);
        return ans ;
    }
};