二叉树的路径和

来源:互联网 发布:淘宝怎么货到付款手机 编辑:程序博客网 时间:2024/05/29 02:15

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

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

样例

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

     1    / \   2   4  / \ 2   3

返回:

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

]

解题思路:直接套用二叉树路径的代码进行一次特判是否加和是目标值即可。

class Solution {public:    /**     * @param root the root of binary tree     * @param target an integer     * @return all valid paths     */    vector<vector<int>>Ans;    vector<int>tem;    void F(vector<int>x,int ans){      int sum =0;      for(int i=0;i<x.size();i++)sum+=x[i];      if(sum==ans)Ans.push_back(x);    }    vector<vector<int>> binaryTreePathSum(TreeNode *root, int target) {        // Write your code here  if(root==NULL)return Ans;  tem.push_back(root->val);  if(root->left!=NULL)binaryTreePathSum(root->left,target);  if(root->right!=NULL)binaryTreePathSum(root->right,target);  if(root->left==NULL&&root->right==NULL){        F(tem,target);  }  tem.erase(tem.begin()+tem.size()-1);  return Ans;    }};


1 0
原创粉丝点击