113.Path Sum II 树的指定和路径

最新推荐文章于 2021-12-31 22:26:10 发布

原创最新推荐文章于 2021-12-31 22:26:10 发布 · 92 阅读

0 ·

CC 4.0 BY-SA版权

本文介绍了一种算法，用于在二叉树中寻找所有根节点到叶子节点的路径，其中路径上的节点值之和等于给定的目标值。通过递归遍历树结构，将当前节点值与目标值比较，如果相等且到达叶子节点，则保存该路径。

题目要求查找是否存在根节点到子节点的路径和等于指定的值，若存在则将路径存入容器。

思路：从根节点开始，将自身值存入临时容器，并开始检索其子节点，将临时容器传递下去；直到叶子节点，判断路径和是否等于指定值，若等于，则保存临时容器的值。

代码：

vector<vector<int>> path;

bool travel(TreeNode* node,int sum,vector<int> temp)
{
    int val=node->val;
    if(node->left==NULL && node->right==NULL)
    {
        if(val==sum)
        {
            temp.push_back(node->val);
            path.push_back(temp);
        }
    }
    //因为不是地址或指针传递，所以传入左右子节点的temp不再是同一个。
    if(node->left!=NULL)
    {
        travel(node->left, sum-val,temp);
    }
    if(node->right!=NULL)
    {
        travel(node->right, sum-val,temp);
    }
    return 0;
}
vector<vector<int>> pathSum(TreeNode* root, int sum) {
    if(root!=NULL)
    {
        vector<int> temp;
        travel(root, sum,temp);
    }
    return path;
}