码迷,mamicode.com
首页 > 其他好文 > 详细

Path Sum II

时间:2014-08-16 22:24:21      阅读:212      评论:0      收藏:0      [点我收藏+]

标签:style   blog   color   io   for   ar   div   amp   

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]
]

思路:对二叉树进行遍历即可。

class Solution {
public:
    vector<vector<int>> pathSum( TreeNode *root, int sum ) {
        vector<int> path;
        PathSumSub( path, root, sum );
        return pathes;
    }
private:
    void PathSumSub( vector<int>& path, TreeNode *node, int sum ) {
        if( !node ) { return; }
        path.push_back( node->val );
        sum -= node->val;
        if( !node->left && !node->right ) {
            if( sum == 0 ) { pathes.push_back( path ); }
            path.pop_back();
            return;
        }
        if( node->left ) { PathSumSub( path, node->left, sum ); }
        if( node->right ) { PathSumSub( path, node->right, sum ); }
        path.pop_back();
    }
    vector<vector<int>> pathes;
};

 

Path Sum II,布布扣,bubuko.com

Path Sum II

标签:style   blog   color   io   for   ar   div   amp   

原文地址:http://www.cnblogs.com/moderate-fish/p/3916949.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!