本文主要是介绍数据结构学习 jz34 二叉树中和为某一值的路径,希望对大家解决编程问题提供一定的参考价值,需要的开发者们随着小编来一起学习吧!
关键词:回溯 二叉树 前序遍历 路径记录
因为我没有仔细接触过二叉树的遍历过程,所以我是懵懵懂懂按照dfs的方法写的。没想到写对了,看了解答发现这叫做二叉树的前序遍历。用时29min。
这让我明白了前序遍历和dfs原来是有相同之处的。(我甚至想按照习惯给它剪枝,后来发现不太行,每条路都必须走一遍才行,我似乎又懂得了许多呢!)
题目:
思路:
注意这里是每个节点都要访问到的,不能剪枝!!!
容器:
得弄两个vector。
std::vector<std::vector<int>> res:记录最后要返回的结果。
std::vector<int> temp:记录中途的路径,如果有符合的路径,就把这个vector push到res里面。
sum记录中途的结果。
注意:函数形参里的std::vector<int>& temp不要漏引用,不然内存会多用很多!
中止条件:
到达叶子节点就停: (别剪枝)
if (root->left == nullptr && root->right == nullptr)
逐个查询:
这里有两个要查询的东西,左节点和右节点。
if (root->left != nullptr)dfs(root->left, target, temp, sum + root->val);
if (root->right != nullptr)dfs(root->right, target, temp, sum + root->val);
复杂度计算:
时间复杂度O(n)
空间复杂度O(n) 主要是统计temp用掉的额外空间,如果数退化成链表,就需要n
代码:
#include <vector>
struct TreeNode {int val;TreeNode *left;TreeNode *right;TreeNode() : val(0), left(nullptr), right(nullptr) {}TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {}};class Solution {
public:std::vector<std::vector<int>> res;std::vector<std::vector<int>> pathTarget(TreeNode* root, int target) {if (root == nullptr) return res;std::vector<int> temp;dfs(root, target, temp, 0);return res;}void dfs(TreeNode* root, int target, std::vector<int>& temp, int sum){temp.push_back(root->val);if (root->left == nullptr && root->right == nullptr)//如果叶子节点{if (sum + root->val == target)//如果符合{res.push_back(temp);}}else{if (root->left != nullptr)dfs(root->left, target, temp, sum + root->val);if (root->right != nullptr)dfs(root->right, target, temp, sum + root->val);}temp.pop_back();//回溯return;}};
这篇关于数据结构学习 jz34 二叉树中和为某一值的路径的文章就介绍到这儿,希望我们推荐的文章对编程师们有所帮助!