题目
给定一个二叉树,返回所有从根节点到叶子节点的路径。
说明: 叶子节点是指没有子节点的节点。
示例
输入:
1
/ \
2 3
\
5
输出:
[“1->2->5”, “1->3”]
解释:
所有根节点到叶子节点的路径为: 1->2->5, 1->3
代码
#include <stdio.h>
#include <iostream>
#include <vector>
#include <cstring>
using namespace std;
struct TreeNode {
int val;
TreeNode *left;
TreeNode *right;
TreeNode(int x) : val(x), left(NULL), right(NULL) {}
};
class Solution {
public:
vector<string> binaryTreePaths(TreeNode* root) {
vector<string> ans;
string s;
dfs(s,root,ans);
return ans;
}
void dfs(string s,TreeNode* root,vector<string>& ans){
if(root == nullptr)
return ;
s += to_string(root->val);
if(root->left==nullptr&&root->right==nullptr){
ans.push_back(s);
return ;
}else{
s += "->";
dfs(s,root->left,ans);
dfs(s,root->right,ans);
}
}
};
int main()
{
TreeNode root(1);
TreeNode r1(2);TreeNode r2(3);TreeNode r3(5);
root.left = &r1;
root.right= &r2;
r1.right = &r3;
Solution s;
vector<string> ans = s.binaryTreePaths(&root);
for(int i=0;i<ans.size();i++)
cout<<ans[i]<<endl;
return 0;
}
今天也是爱zz的一天哦!