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