[leetcode]94.二叉树中序遍历

给定一个二叉树,返回它的中序 遍历。

示例:

输入: [1,null,2,3]
   1
    \
     2
    /
   3

输出: [1,3,2]

进阶: 递归算法很简单,你可以通过迭代算法完成吗?

思路1:递归法(标准模板)

class Solution {
   public:
    vector<int> ans;
    vector<int> inorderTraversal(TreeNode* root) {
        if (root != NULL) {
            inorderTraversal(root->left);
            ans.push_back(root->val);
            inorderTraversal(root->right);
        }
        return ans;
    }
};

思路2:迭代法(二叉树的非递归遍历)
1.定义一个栈st,根节点入栈。
2.如果栈顶元素存在左孩子,则一直向左,左孩子入栈,直到不存在左孩子的节点。
3.如果当前节点不存在左孩子,栈顶元素出栈压入答案数组,右孩子入栈,继续回到2判断,直至栈空。

AC代码:(C++):

    vector<int> inorderTraversal(TreeNode* root) {
        vector<int> ans;
        stack<TreeNode*> st;
        TreeNode* temp = root;
        while (temp || !st.empty()) {
            while (temp) {  //如果存在左孩子,则一直向下入栈
                st.push(temp);
                temp = temp->left;
            }
            temp = st.top();  //取栈顶并出栈
            st.pop();
            ans.push_back(temp->val);
            temp = temp->right;  //如果存在右孩子,右孩子入栈,继续判断
        }
        return ans;
    }

你可能感兴趣的:(LeetCode)