二叉树的路径和

来源:互联网 发布:c 高级编程第10版 编辑:程序博客网 时间:2024/06/05 02:30

一、问题描述

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

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

二、样例

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

     1    / \   2   4  / \ 2   3

返回:

[  [1, 2, 2],  [1, 4]]
三、思路

对每条路径进行加和,与给定的值相比,如果相等就加到向量中。

四、代码

/**
 * 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>> binaryTreePathSum(TreeNode *root, int target) {
      vector<vector<int> >paths;
        vector<int>path;
        vector<int>::iterator it;
        if(root==NULL){
            return paths;
        }
        else {
            def(root,paths,path,target);
            return paths;
        }
    }
    void def(TreeNode*root,vector<vector<int> >&ps,vector<int>p,int target){
        p.push_back(root->val);
        vector<int>::iterator itp;
        if(root->left==NULL&&root->right==NULL){
                int flag=0;
            for(itp=p.begin();itp!=p.end();itp++){
                flag+=*itp;
            }
            if(flag==target){
                ps.push_back(p);
            }
        }
        if(root->left!=NULL){
            def(root->left,ps,p,target);
        }
        if(root->right!=NULL){
            def(root->right,ps,p,target);
        }
        // Write your code here
    }
};

0 0
原创粉丝点击