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 and sum = 22,

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

return

[   [5,4,11,2],   [5,8,4,5]]

二.题目分析

这道题与Path Sum类似，不同的是这道题要求找到所有和等于给定值的路径，必须遍历完所有的路径。在找到满足的路径之后，不能直接返回，而是将其添加到一个vector<vector<int>>中。在查找的过程中，每经过一个结点，先使用一个vector<int>将该路径中的所有结点记录下来。需要注意的是，在进入每一个结点的时候，先将该结点的值push到vector中，在退出时将该结点的值pop出来，这样就可以避免有时会忘记pop结点的值的情况。

该方法的时间复杂度为O(n)，空间复杂度为O(logn)。

三.示例代码

struct TreeNode{    int val;    TreeNode *left;    TreeNode *right;    TreeNode(int x) : val(x), left(NULL), right(NULL) {}};class Solution{public:    vector<vector<int>> pathSum(TreeNode* root, int sum)    {        vector<vector<int> > Result;        vector<int> TmpResult;        pathSum(root, sum, Result, TmpResult);        return Result;    }private:    void pathSum(TreeNode* root, int sum, vector<vector<int> > &Result, vector<int> &TmpResult)    {        if (!root)            return;        if (!root->left && !root->right && root->val == sum)        {            TmpResult.push_back(sum);            Result.push_back(TmpResult);            // pop the leaf node              TmpResult.pop_back();            return;        }        int SumChild = sum - root->val;        TmpResult.push_back(root->val);        pathSum(root->left, SumChild, Result, TmpResult);        pathSum(root->right, SumChild, Result, TmpResult);        TmpResult.pop_back();    }};

四.小结

与Path Sum一样，用DFS解决。