首页 > 代码库 > [C++]LeetCode: 91 Path Sum II

[C++]LeetCode: 91 Path Sum II

2024-11-02 03:13:02   205人阅读

题目：

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类似，当当前节点是叶子结点，并且当前节点值等于sum值，先把当前节点值push进tmp，再返回。如果当前节点非空，将当前结果压进tmp中，再接下来递归其左子树和右子树，寻找所有的可能组合。

Attention:

1. 注意只有存在左右孩子结点时，才递归调用辅助函数，搜索左右子树。

//需要判断是否有左右结点，没有就不递归调用了
            if(root->left) pathSum_helper(root->left, newSum, tmp, ret);
            if(root->right) pathSum_helper(root->right, newSum, tmp, ret);

2. 当判断当前节点符合迭代终止条件后，首先应先把当前节点值push进数组，再返回结果。（上一层调用中并没有push当前节点值）

/如果此时的节点就是叶子节点，并且符合sum条件，先push进tmp，再返回。
        if(!root->left && !root->right && root->val == sum)
        {
            tmp.push_back(root->val);
            ret.push_back(tmp);
            return;
        }
AC Code：

/**
 * Definition for binary tree
 * 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> > ret;
        vector<int> tmp;
        if(root == NULL) return ret;
        pathSum_helper(root, sum, tmp, ret);
        return ret;
    }
    
private:
    void pathSum_helper(TreeNode* root, int sum, vector<int> tmp, vector<vector<int>>& ret)
    {
        //如果此时的节点就是叶子节点，并且符合sum条件，先push进tmp，再返回。
        if(!root->left && !root->right && root->val == sum)
        {
            tmp.push_back(root->val);
            ret.push_back(tmp);
            return;
        }
       
        if(root != NULL)
        {
            tmp.push_back(root->val);
            int newSum = sum - root->val;
            //需要判断是否有左右结点，没有就不递归调用了
            if(root->left) pathSum_helper(root->left, newSum, tmp, ret);
            if(root->right) pathSum_helper(root->right, newSum, tmp, ret);
        }
        
        return;
    }
};



[C++]LeetCode: 91 Path Sum II


联系
我们