day17
思路:
前序遍历,因为要从根节点开始找。
回溯:
返还到调用函数之前的状态
先来看参数
path存储的是 单条路径,也就是从根结点到一个叶子结点的路径。
当到了叶子结点以后,就把路径插入到容器中。然后返回。遍历到最后一个叶子结点的时候,也就是递归结束的时候。
返回到上一层的时候,需要思考,有没有因为本层的递归而给上一层带来了变化,影响了上一层的结果。
在本题:
带来的变化只能有一个,res 中增加了一条路径。
所以当返回的时候,path要和之前调用的一样。这样才能从该结点继续向别的结点遍历。
void traversal(TreeNode* root,string path,vector<string>& res)
这里的参数,path。用的是值传递。也就是上一层递归和下一层递归之间的path是没有联系的。所以当函数返回到上一层的时候,用的就是上一层之前的path。
这里的res 用的是引用传递。为的就是在一层层递归中,res的值不会“消失”。
而path是每条路径都不一样,所以用值传递。
再来看函数调用
if(root->left) traversal(root->left, path + "->", res);
if(root->right) traversal(root->right, path + "->", res);
这里就有回溯,因为这里的 " -> " 是直接在函数中的,所以如果返回到上一层的话,就直接会被清空。
如果 " -> " 不写在函数中,也就是直接写在外面
if (cur->left) {
path += "->";
traversal(cur->left, path, result); // 左
}
if (cur->right) {
path += "->";
traversal(cur->right, path, result); // 右
}
这样的话,每次返回到本层的时候,如果从左边到右边的时候, 会发现前面多加了 " -> " 。
因为改变了 path 的值
所以需要手动pop掉
if (cur->left) {
path += "->";
traversal(cur->left, path, result); // 左
path.pop_back(); // 回溯 '>'
path.pop_back(); // 回溯 '-'
}
if (cur->right) {
path += "->";
traversal(cur->right, path, result); // 右
path.pop_back(); // 回溯 '>'
path.pop_back(); // 回溯 '-'
}
注意点:
1. 将 val 转为字符串
2. 将 “中” 放在前面。因为如果是叶子结点的话,需要先将叶子结点插入,再弹出。
3. 为了代码精简,可以将 "->" 放在函数的参数中,这样就不用手动的pop元素。
4.为了防止空结点操作,在调用之前都需要判断结点是否为空
class Solution {
private:
void traversal(TreeNode* cur, string path, vector<string>& result) {
path += to_string(cur->val); // 中,中为什么写在这里,因为最后一个节点也要加入到path中
if (cur->left == NULL && cur->right == NULL) { // 直接处理叶子
result.push_back(path);
return;
}
if (cur->left) {
path += "->";
traversal(cur->left, path, result); // 左
}
if (cur->right) {
path += "->";
traversal(cur->right, path, result); // 右
}
}
public:
vector<string> binaryTreePaths(TreeNode* root) {
vector<string> result;
string path;
if (root == NULL) return result;
traversal(root, path, result);
return result;
}
};