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;
    }
};