题目要求:要求输出一个二叉树的所有从根节点到叶子节点的路径。
解题方法:
1. 可以用递归的方法实现,将根节点的左右子树所输出的两个数组合并为一个数组,并将得到的数组的每一个string元素头部加上"to_string(root->va)->l",其中to_string()是C++中的一个可以将int型转换为string类型的函数。
/**
* 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:
vector<string> binaryTreePaths(TreeNode* root) {
vector<string> vs;
if(root==NULL)
return vs;
if(root->left==NULL&&root->right==NULL){
vs.push_back(to_string(root->val));
return vs;
}
if(root->left!=NULL)
vs=binaryTreePaths(root->left);
if(root->right!=NULL){
vector<string> temp;
temp=binaryTreePaths(root->right);
for(int i=0;i<temp.size();i++){
vs.push_back(temp[i]);
}
}
for(int i=0;i<vs.size();i++){
vs[i]=to_string(root->val)+"->"+vs[i];
}
return vs;
}
};
2. 用循环的方式来实现,因为有回溯的过程,经过那些既有左子节点又有右子节点的那些节点均有两条路径,需要回溯,所以需要用到两个栈,一个栈st用来存储既有左子节点又有右子节点的那些节点,另一个栈用来存储相应的从根节点到该节点的路径。
/**
* 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:
vector<string> binaryTreePaths(TreeNode* root) {
stack<TreeNode*> st;
stack<string> ss;
vector<string> vs;
if(root==NULL)
return vs;
if(root->left==NULL&&root->right==NULL){
vs.push_back(to_string(root->val));
return vs;
}
TreeNode* p=root;
string tem=to_string(p->val);
while(p){
if(p->left&&p->right){
st.push(p);
ss.push(tem);
p=p->left;
tem=tem+"->"+to_string(p->val);
}
else if(p->left){
p=p->left;
tem=tem+"->"+to_string(p->val);
}
else if(p->right){
p=p->right;
tem=tem+"->"+to_string(p->val);
}
else
break;
}
vs.push_back(tem);
while(!st.empty()){
p=st.top();
st.pop();
tem=ss.top();
ss.pop();
p=p->right;
tem=tem+"->"+to_string(p->val);
while(p){
if(p->left&&p->right){
st.push(p);
ss.push(tem);
p=p->left;
tem=tem+"->"+to_string(p->val);
}
else if(p->left){
p=p->left;
tem=tem+"->"+to_string(p->val);
}
else if(p->right){
p=p->right;
tem=tem+"->"+to_string(p->val);
}
else
break;
}
vs.push_back(tem);
}
return vs;
}
};