LC-二叉树中是否存在节点和为指定值的路径

最新推荐文章于 2021-10-10 10:34:32 发布

vector6_

最新推荐文章于 2021-10-10 10:34:32 发布

阅读量140

点赞数

分类专栏： Leetcode 文章标签：数据结构算法二叉树

本文链接：https://blog.csdn.net/qq_41105501/article/details/109345137

版权

Leetcode 专栏收录该内容

137 篇文章 0 订阅

订阅专栏

二叉树中是否存在节点和为指定值的路径

//二叉树中是否存在节点和为指定值的路径
/*给定一个二叉树和一个值sum，
请找出所有的根节点到叶子节点的节点值之和等于sum的路径*/
#include<vector>
using namespace std;
struct TreeNode {
	int val;
	struct TreeNode *left;
	struct TreeNode *right;
};


class Solution {
public:
	/**
	*
	* @param root TreeNode类
	* @param sum int整型
	* @return int整型vector<vector<>>
	*/
	//引用及非引用形参的使用解决了递归中res的回退问题
	//res每次通过函数调用，会复制一个新的vector存放数据
	void dfs(TreeNode* root, int sum, vector<int> res, vector<vector<int>>& result)
	{
		if (root == nullptr)
			return;
		res.push_back(root->val);
		if (root->val == sum && root->left == nullptr && root->right == nullptr)
		{
			result.push_back(res);
		}
		dfs(root->left, sum - root->val, res, result);

		dfs(root->right, sum - root->val, res, result);
	}
	vector<vector<int> > pathSum(TreeNode* root, int sum) {
		vector<vector<int>> result;
		vector<int> res;
		dfs(root, sum, res, result);
		return result;
	}
};