二叉树的路径和

来源:互联网 发布:非诚勿扰网络直播时间 编辑:程序博客网 时间:2024/06/06 01:00

描述:

    给定一个二叉树,找出所有路径中各节点相加总和等于给定 目标值 的路径。


一个有效的路径,指的是从根节点到叶节点的路径。


样例:给定一个二叉树,和 目标值 = 5:


     1
    / \
   2   4
  / \
 2   3
返回:


[
  [1, 2, 2],
  [1, 4]
]

实现思路:

       就像之前写的所有路径,我们需要找到符合条件的路径,就是需要把每条路径的和算出来,如果等于指定值我们就把他加入到va向量里面。但是,每次不要忘了删除上一条路径。

实现代码:

 /**
 * Definition of TreeNode:
 * class TreeNode {
 * public:
 *     int val;
 *     TreeNode *left, *right;
 *     TreeNode(int val) {
 *         this->val = val;
 *         this->left = this->right = NULL;
 *     }
 * }
 */
class Solution {
public:
    /**
     * @param root the root of binary tree
     * @param target an integer
     * @return all valid paths
     */
    vector<vector<int>> va;  
    void pathsum(TreeNode *root,int sum,int goal,vector<int> v)  
    {  
          
        if(root->left==NULL && root->right==NULL)  
        {  
            if(sum==goal)  
            va.push_back(v);  
            return;  
        }  
          
        int s;  
          
        if(root->left!=NULL)  
        {  
            vector<int> vm(v);  
            s=sum+root->left->val;  
            vm.push_back(root->left->val);  
            pathsum(root->left,s,goal,vm);  
        }  
        if(root->right!=NULL)  
        {  
            vector<int> vm(v);  
            s=sum+root->right->val;  
            vm.push_back(root->right->val);  
            pathsum(root->right,s,goal,vm);  
        }  
    }  
    vector<vector<int>> binaryTreePathSum(TreeNode *root, int target) {  
        // Write your code here  
        if(root==NULL)  
        return va;  
        vector<int>v;  
        v.push_back(root->val);  
        pathsum(root,root->val,target,v);  
        return va; 
    }
};

做题感想:通过这个题,进一步体会vector向量的用法。进一步体会pop_back的用法。


0 0
原创粉丝点击