【leetcode】94Binary Tree Inorder Traversal(非递归中序遍历二叉树)

题目大意:给出一棵二叉树,要求用非递归的方法返回中序遍历的结点值的序列


递归形式的中序遍历是先中序遍历完左子树后再访问根节点,然后遍历右子树

一种借助栈来非递归的实现中序遍历过程如下:

第一种方法:

这个方法比较容易理解,但缺点是会改变原本的树的结构,在确定栈顶节点有左孩子后,将其左孩子入栈,然后将栈顶节点左孩子置为空,这样等遍历完该节点的左子树再次访问该节点时就不会因为有左孩子而无法跳出循环。

只需要0ms

/**
 * 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 inorderTraversal(TreeNode* root) {
        vector ans;
        stack S;
        if(root) S.push(root);
        while(!S.empty())
        {
            TreeNode *node=S.top();
            if(node->left) {S.push(node->left);node->left=NULL;}
            else
            {
                ans.push_back(node->val);
                S.pop();
                if(node->right) S.push(node->right);
            }
        }
        return ans;
    }
};


第二种方法:在第一种基础上进行了一些改变,不需要改变树的结构,但是没有第一个效率高,运行时间4ms,之后再看怎么提高效率吧……

class Solution {
public:
    vector inorderTraversal(TreeNode* root) {
        vector ans;
        stack S;
        TreeNode *current=root;
        while(!S.empty() || current)
        {
            if(current)
            {
                S.push(current);
                current=current->left;
            }
            else
            {
                TreeNode *tmp=S.top();
                ans.push_back(tmp->val);
                S.pop();
                current=tmp->right;
            }
        }
        return ans;
    }
};


你可能感兴趣的:(leetcode)