您的位置:首页 > 其它

Leetcode_Path Sum II

2015-02-09 09:53 316 查看
题意需要记录所有可能的路径,可以设置一个变量path用于存储每次产生的路径,设置一个变量result存储所有符合条件的路劲。

用递归可以实现,主要注意的问题是:对path路径的pop操作。

方式一:每一次深度遍历之后,得到一条路径,符合条件就压入result中,然后要判断下一天路径。下一条路径是将前一条路径中的最后一个左节点弹出,压入右节点。下面这种方式,就是在寻得叶子节点之后,继续执行叶子节点的左节点和右节点的遍历,两种都是return,之后执行的pop操作,就可以弹出最后压入的叶子节点。

class Solution {
public:
vector<vector<int> > pathSum(TreeNode *root, int sum) {
vector<vector<int> > result;
vector<int> path;
if(!root) return result;
getPath(result,path,root,sum);
return result;
}
private:
void getPath(vector<vector<int> > &result,vector<int> &path,TreeNode *root,int sum)
{
if(!root)
return ;
path.push_back(root->val);
if(root->left ==nullptr&&root->right==nullptr&&root->val==sum)
{
result.push_back(path);
//return;
}
getPath(result,path,root->left,sum-root->val);
//path.pop_back();
getPath(result,path,root->right,sum-root->val);
path.pop_back();
}
};方式二:
单独处理叶子节点,遇到叶子节点则不再往下遍历,将其pop出,继续下一次右节点的遍历

class Solution {
public:
vector<vector<int> > pathSum(TreeNode *root, int sum) {
vector<vector<int> > result;
vector<int> path;
if(!root) return result;
getPath(result,path,root,sum);
return result;
}
private:
void getPath(vector<vector<int> > &result,vector<int> &path,TreeNode *root,int sum)
{
if(!root)
return ;
path.push_back(root->val);
if(root->left ==nullptr&&root->right==nullptr&&root->val==sum)
{
result.push_back(path);
path.pop_back();
return;
}
if(root->left ==nullptr&&root->right==nullptr&&root->val!=sum)
{
path.pop_back();
return;
}
getPath(result,path,root->left,sum-root->val);
//path.pop_back();
getPath(result,path,root->right,sum-root->val);
path.pop_back();
}
};
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签:  Leetcode Path Sum II