关键词:回溯 二叉树 前序遍历 路径记录
因为我没有仔细接触过二叉树的遍历过程,所以我是懵懵懂懂按照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;}};