前序遍历二叉树递归解法

2013-11-20 14:24:11 · 作者: · 浏览: 156
    Binary Tree Preorder Traversal
    Given a binary tree, return the preorder traversal of its nodes' values.
    For example:
    Given binary tree {1,#,2,3},
    1
    \
    2
    /
    3
    return [1,2,3].
    Note: Recursive solution is trivial, could you do it iteratively
    的确如note说的一样,使用递归只需要几分钟就解决了,如果不是用递归,自己琢磨的话还是花了差不多半个小时。
    看代码就知道非递归比递归长很多。
    非递归方法主要难点:
    1. 需要使用两个循环,左子树循环嵌套右子树循环;
    2. 有右子树的时候,需要跳出本循环,让其重新进入左子树循环。
    2. 无左子树的时候,需要出栈,而不是等到空子树才出栈。
    下面使用两种方法解决一下这个题目,他们的时间效率都差不多,都是12ms,大概测试数据都很少,所以时间都一样。也是LeetCode上的一个水题了,测试数据都不多。
    主函数都是一样的:
    [cpp]
    vector<int> preorderTraversal(TreeNode *root) {
    vector<int> prePath;
    preStack(root, prePath);
    return prePath;
    }
    递归法:
    [cpp]
    void preNodes(TreeNode *node,vector<int> &onePath)
    {
    if(node == nullptr) return;
    onePath.push_back(node->val);
    preNodes(node->left, onePath);
    preNodes(node->right, onePath);
    }
    非递归法:
    [cpp]
    void preStack(TreeNode *node, vector<int> &prePath)
    {
    if(node == nullptr) return;
    stack<TreeNode *> stk;
    TreeNode *cur = node;
    stk.push(cur);
    while (!stk.empty())
    {
    cur = stk.top();
    prePath.push_back(cur->val);
    while (cur->left)
    {
    prePath.push_back(cur->left->val);
    cur = cur->left;
    stk.push(cur);
    }
    stk.pop();
    while (!stk.empty() || cur->right)
    {
    if(cur->right)
    {
    cur = cur->right;
    stk.push(cur);
    break;
    }
    else
    {
    cur = stk.top();
    stk.pop();
    }
    }//while
    }//while
    }