首页 > 其他 > 详细

LeetCode:Path Sum II

时间:2014-11-26 11:17:58      阅读:239      评论:0      收藏:0      [点我收藏+]

题目描述:

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

代码:

void search_path(TreeNode * node,int sum,vector<int> temp,vector<vector<int> > &result)
{
    if(node == NULL)
        return;
    temp.push_back(node->val);
    if(node->left == NULL && node->right == NULL)
    {
        int current_sum = 0;
        for(int i = 0;i < temp.size();i++)
            current_sum = current_sum + temp[i];
        if(sum == current_sum)
            result.push_back(temp);
        return;
    }
    if(node->left != NULL)
        search_path(node->left,sum,temp,result);
    if(node->right != NULL)
        search_path(node->right,sum,temp,result);
}

vector<vector<int> > pathSum(TreeNode *root, int sum)
{
    vector<vector<int> > result;
    TreeNode * node = root;
    vector<int> temp;
    search_path(node,sum,temp,result);
    return result;
}


LeetCode:Path Sum II

原文:http://blog.csdn.net/yao_wust/article/details/41512503

(0)
(0)
   
举报
评论 一句话评论(0
关于我们 - 联系我们 - 留言反馈 - 联系我们:wmxa8@hotmail.com
© 2014 bubuko.com 版权所有
打开技术之扣,分享程序人生!