题目链接:
写在之前:
这个题目的思路比较清晰,采用递归方式去遍历叶节点即可,即深度优先搜索(DFS),比较难处理的就是如何构造结果集,换句话说就是如何将已经遍历的节点保存起来,并将已遍历完的路径按要求方式添加到结果数组中。
我们采用的方式是:利用函数参数传递的方式,将正在遍历的路径使用string变量保存,并传递给下一次函数调用,遍历完一条路径后添加到vector中。
题目描述:
Given a binary tree, return all root-to-leaf paths.
For example, given the following binary tree:
1
/ \
2 3
\
5All root-to-leaf paths are:
[“1->2->5”, “1->3”]
给定一颗二叉树,返回所有从根节点到叶节点的路径。
题目解释:
从根节点向下到某一个叶节点叫做一条路径,即深度优先搜索产生的结果。
解题方案:
很明显,递归是最先想到的解决方案。
先访问根节点,然后再访问根节点的左子树和右子树,不断递归的处理,直到访问到叶节点(无节点可以访问了)就是一条路径。
那么如何保存已经遍历过的路径呢?
我们采用函数参数传递的方式:
void dfs(TreeNode *node, string path, vector<string>& result)
string path代表的就是我们当前正在遍历的路径,遍历完一个节点后,我们将当前节点添加到path中:
path += to_string(node->val);
然后递归调用时:
dfs(node->left, path + "->", result);
总的结果集使用引用参数传递。
这就是解决方案,在这里有一个比较闹心的事情,刚开始做的时候,我对C++的STL不是很熟,并不知道有to_string()这个函数,因为节点的数据val的类型是int型,路径要求是string类型,所以我在int转string中浪费了很多时间,最后发现STL有一个现成的函数to_string()直接支持这种转换。
下面看AC代码:
代码实现L
/**
* Definition for a binary tree node.
* struct TreeNode {
* int val;
* TreeNode *left;
* TreeNode *right;
* TreeNode(int x) : val(x), left(NULL), right(NULL) {}
* };
*/
class Solution {
public:
/**
*deepth-first search the path
*/
void dfs(TreeNode *node, string path, vector<string>& result)
{
path += to_string(node->val);
//遍历到叶节点的时候代表路径终结,将新路径添加到结果集中
if(node->left == NULL && node->right == NULL)
{
result.push_back(path);
return ;
}
if(node->left != NULL)
{
dfs(node->left, path + "->", result);
}
if(node->right != NULL)
{
dfs(node->right, path + "->", result);
}
return ;
}
//get the path from root to leaf
vector<string> binaryTreePaths(TreeNode* root)
{
vector<string> result;
if(root == NULL)
{
return result;
}
string path;
dfs(root, path, result);
return result;
}
};