Lintcode 二叉树的前序遍历 (递归和非递归法)

给出一棵二叉树,返回其节点值的前序遍历。

样例

给出一棵二叉树 {1,#,2,3},

   1
    \
     2
    /
   3

 返回 [1,2,3].

挑战 

你能使用非递归实现么?



标签 

递归  二叉树  二叉树遍历  非递归

一、递归形式:

/**
 * Definition of TreeNode:
 * class TreeNode {
 * public:
 *     int val;
 *     TreeNode *left, *right;
 *     TreeNode(int val) {
 *         this->val = val;
 *         this->left = this->right = NULL;
 *     }
 * }
 */

class Solution {
public:
    /**
     * @param root: The root of binary tree.
     * @return: Preorder in vector which contains node values.
     */
     void pre(vector &result,TreeNode *root){
         result.push_back(root->val);
         if(root->left!=NULL)
         pre(result,root->left);
         if(root->right!=NULL)
         pre(result,root->right);
     }
    vector preorderTraversal(TreeNode *root) {
        // write your code here
        vector  result;
        if(root==NULL)
        return result;
        pre(result,root);
        return result;
    }
};

二、非递归形式:

使用栈来记录没有遍历右子节点的位置。

/**
 * Definition of TreeNode:
 * class TreeNode {
 * public:
 *     int val;
 *     TreeNode *left, *right;
 *     TreeNode(int val) {
 *         this->val = val;
 *         this->left = this->right = NULL;
 *     }
 * }
 */

class Solution {
public:
    /**
     * @param root: The root of binary tree.
     * @return: Preorder in vector which contains node values.
     */
    vector preorderTraversal(TreeNode *root) {
        // write your code here
        vector result;
        stack s;
        if(root==NULL)
        return result;
        while(1){
            if(root!=NULL)
            result.push_back(root->val);
            if(root->right!=NULL)
            s.push(root);
            if(root->left!=NULL)
            root=root->left;
            else if(!s.empty())
            {
            root=s.top()->right;
            s.pop();
            }
            else
            break;
            
        }
        
    }
};

如有问题请留言。

如有帮助请顶一个,你们的支持是我最大的动力。

文章均可以转载,但请注明文章链接,谢谢。



你可能感兴趣的:(LintCode编程)