LintCode 67. 二叉树的中序遍历

67二叉树的中序遍历

描述

给出一棵二叉树,返回其中序遍历

样例

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

   1
    \
     2
    /
   3

返回 [1,3,2]

题目链接:https://www.lintcode.com/problem/binary-tree-inorder-traversal/description

解题思路1:(递归)递归得到左子树的中序遍历的vector数组,把当前结点值添加进数组,再递归得到右子树中序遍历的vector数组,合并两个vector数组并返回。结点为NULL时即遍历到最底层,直接返回空的vector数组。

class Solution {
public:
    /**
     * @param root: A Tree
     * @return: Inorder in ArrayList which contains node values.
     */
    vector<int> inorderTraversal(TreeNode * root) {
        // write your code here
        vector<int> ans;
        if(root == NULL)
            return ans;
        if(root->left != NULL){
            vector<int> leftans = inorderTraversal(root->left);
            ans.insert(ans.end(),leftans.begin(),leftans.end());
        }
        ans.push_back(root->val);
        if(root->right != NULL){
            vector<int> rightans = inorderTraversal(root->right);
            ans.insert(ans.end(),rightans.begin(),rightans.end());
        }
        return ans;
    }
};

解题思路2:(非递归)使用栈作为节点的存储结构,从root节点访问,首先把root节点压入栈,然后遍历栈,每次访问栈顶节点。如果节点的左节点非空,把左节点压入栈并把左节点标记为NULL,然后遍历下一个节点。如果左节点为空,则把当前节点放入vector数组,栈节点pop掉,再判断右节点,如果右节点非空,则把右节点压入栈并把右节点标记为NULL,遍历下一个节点。

class Solution {
public:
    /**
     * @param root: A Tree
     * @return: Inorder in ArrayList which contains node values.
     */
    vector<int> inorderTraversal(TreeNode * root) {
        // write your code here
        vector<int> ans;
        if(root == NULL)
            return ans;
        stack <TreeNode*> st;
        st.push(root);
        while(st.size()>0){
            TreeNode* cur = st.top();
            if(cur->left != NULL){
                st.push(cur->left);
                cur -> left = NULL;
                continue; 
            }
            ans.push_back(cur->val);
            st.pop();
            if(cur->right != NULL){
                st.push(cur->right);
                cur -> right = NULL;
            }
        }
        return ans;
    }
};



阅读更多
版权声明:本文为博主原创文章,未经博主允许不得转载。 https://blog.csdn.net/CriminalCode/article/details/80350392
个人分类: lintcode
想对作者说点什么? 我来说一句
相关热词

没有更多推荐了,返回首页

关闭
关闭
关闭