leetcode 113 —— Path Sum II

最新推荐文章于 2020-07-15 10:36:06 发布

原创最新推荐文章于 2020-07-15 10:36:06 发布 · 316 阅读

0 ·

CC 4.0 BY-SA版权

算法学习同时被 2 个专栏收录

166 篇文章

订阅专栏

数据结构

163 篇文章

订阅专栏

本文介绍了一种使用回溯法寻找二叉树中所有从根节点到叶子节点的路径，这些路径上的节点值之和等于指定目标值的方法。以一个具体的二叉树结构为例，展示了如何找到所有符合条件的路径。

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

思路：回溯法，事实表明，使用引用的路径能显著提高运行速度

class Solution {
public:
	vector<vector<int>> pathSum(TreeNode* root, int sum) {
		vector<vector<int>> res;
		vector<int> path;
		dfs(root, 0, sum, path, res);
		return res;
	}
	void dfs(TreeNode* root, int cnt, int sum, vector<int>& path, vector<vector<int>> &res){
		if (!root)
			return;
		if (!root->left&&!root->right){
			if (cnt + root->val == sum){
				path.push_back(root->val);
				res.push_back(path);
				path.pop_back();
				return;
			}
		}
		path.push_back(root->val);
		dfs(root->left, cnt + root->val, sum, path, res);
		dfs(root->right, cnt + root->val, sum, path, res);
		path.pop_back();
		return;  //其实不加也可以，void函数执行完后自动返回上级
	}
};