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解决。

版权声明：本文为博主原创文章，未经博主允许不得转载。