本文记录的是刷题过程中的重要概念和笔记。如有侵权,请联系删除。
257. 二叉树的所有路径
力扣题目链接(opens new window)
https://leetcode.cn/problems/binary-tree-paths/
给定一个二叉树,返回所有从根节点到叶子节点的路径。
说明: 叶子节点是指没有子节点的节点。
示例: 257.二叉树的所有路径1
思路
递归和回溯就是一家的,本题也需要回溯。
递归
三要素分析:
- 参数返回值
- 终止条件:叶子结点,记录路径
- 单层:找左右
注意:
- 回溯要和递归永远在一起。典型错误:
if (cur->left) {
traversal(cur->left, path, result);
}
if (cur->right) {
traversal(cur->right, path, result);
}
path.pop_back();
- 注意在函数定义的时候void traversal(TreeNode* cur, string path, vector<string>& result) ,定义的是string path,每次都是复制赋值,不用使用引用,否则就无法做到回溯的效果。(这里涉及到C++语法知识)
- 回溯就隐藏在traversal(cur->left, path + “->”, result);中的 path + “->”。 每次函数调用完,path依然是没有加上"->" 的,这就是回溯了。
- 为什么不去定义一个 string& path 这样的函数参数呢,path += to_string(cur->val); 每次是加上一个数字,这个数字如果是个位数,那好说,就调用一次path.pop_back(),但如果是 十位数,百位数,千位数呢? 百位数就要调用三次path.pop_back(),才能实现对应的回溯操作。
traversal 英 / trəˈvərs(ə)l / 美 / trəˈvərs(ə)l / 简明 n. [计] 遍历;横越;横断物
/**
* Definition for a binary tree node.
* 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:
void traversal(TreeNode* node, string path, vector<string>& result){
path += to_string(node->val);
if(!node->left && !node->right){
result.push_back(path);
return;
}
if(node->left) traversal(node->left,path +"->", result);
if(node->right) traversal(node->right,path +"->", result);
return;
}
vector<string> binaryTreePaths(TreeNode* root) {
vector<string> result;
if(root) traversal(root,"",result);
return result;
}
};
补充:vector& result
记得加&,否则是按值传递,最后啥都没有
迭代法
使用前序遍历的迭代方式来模拟遍历路径的过程。用stack存储路径。
存放对应的遍历路径。如果只是存放访问过的节点,那么很有可能一些几点没有办法得到及时的排出,所以导致路径出错,因此呢存储的方式是每一个节点 和 对应的当前的访问路径 各自放到各自的栈中。
class Solution {
public:
vector<string> binaryTreePaths(TreeNode* root) {
stack<TreeNode*> treeSt;// 保存树的遍历节点
stack<string> pathSt; // 保存遍历路径的节点
vector<string> result; // 保存最终路径集合
if (root == NULL) return result;
treeSt.push(root);
pathSt.push(to_string(root->val));
while (!treeSt.empty()) {
TreeNode* node = treeSt.top(); treeSt.pop(); // 取出节点 中
string path = pathSt.top();pathSt.pop(); // 取出该节点对应的路径
if (node->left == NULL && node->right == NULL) { // 遇到叶子节点
result.push_back(path);
}
if (node->right) { // 右
treeSt.push(node->right);
pathSt.push(path + "->" + to_string(node->right->val));
}
if (node->left) { // 左
treeSt.push(node->left);
pathSt.push(path + "->" + to_string(node->left->val));
}
}
return result;
}
};