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

Leetcode_Path Sum II

时间:2015-02-09 11:02:18      阅读:120      评论:0      收藏:0      [点我收藏+]

标签:leetcode   path sum ii   

题意需要记录所有可能的路径,可以设置一个变量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

标签:leetcode   path sum ii   

原文地址:http://blog.csdn.net/yinqiaohua/article/details/43667501

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