您的位置:首页 > 其它

leetcode Path Sum II

2015-10-20 15:33 211 查看
python:

# Definition for a binary tree node.
# class TreeNode(object):
#     def __init__(self, x):
#         self.val = x
#         self.left = None
#         self.right = None

class Solution(object):
def __init__(self):
self.paths = []
def recur  (self, root, path, sum):
if root.left==root.right==None:
if root.val ==sum:
self.paths.append(path+[sum])
return
else:
new_path = path+[root.val]
if root.left!=None:
self.recur(root.left,new_path,sum-root.val)
if root.right!=None:
self.recur(root.right,new_path,sum-root.val)

def pathSum(self, root, sum):
"""
:type root: TreeNode
:type sum: int
:rtype: List[List[int]]
"""
if root==None : return [];
self.recur(root,[],sum)
return self.paths


c++:

/**
* Definition for a binary tree node.
* struct TreeNode {
*     int val;
*     TreeNode *left;
*     TreeNode *right;
*     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
* };
*/
class Solution {
public:

vector<vector<int>> paths;

vector<vector<int>> pathSum(TreeNode* root, int sum) {
vector<int> path;
if (root==NULL) return paths;
recur(root,path,sum);
return paths;
}

void recur(TreeNode* root, vector<int>& path, int sum){
path.push_back(root->val);
if (root->left==NULL && root->right==NULL){
if (root->val==sum) paths.push_back(path);
path.pop_back();
return;
}
else{
if (root->left!=NULL) recur(root->left,path,sum-root->val);
if (root->right!=NULL)recur(root->right,path,sum-root->val);
path.pop_back();
<span style="white-space:pre">	</span>   return;
}
}
};
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签:  leetcode