题目描述
给定一个二叉树,返回它的前序遍历。
示例:
输入: [1,null,2,3]
1
\
2
/
3
输出: [1,2,3]
进阶: 递归算法很简单,你可以通过迭代算法完成吗?
递归算法
代码解答
/**
* 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 {
vector<int> v;
public:
vector<int> preorderTraversal(TreeNode* root) {
if(root){
v.push_back(root->val);
preorderTraversal(root->left);
preorderTraversal(root->right);
}
return v;
}
};
非递归算法
代码解答
class Solution {
stack<TreeNode*> stk;
public:
vector<int> preorderTraversal(TreeNode* root) {
vector<int> vec;
TreeNode* temp = root;
while(temp){
vec.push_back(temp -> val);
if(temp->right){
stk.push(temp->right);
}
if(temp->left){
temp = temp->left;
}
else{
// 防止根节点为空
if(stk.empty()){
break;
}
temp = stk.top();
stk.pop();
}
}
return vec;
}
};