Leetcode刷题笔记 94. 二叉树的中序遍历

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

示例1
输入：

[1,null,2,3]
   1
    \
     2
    /
   3

输出：
[1,3,2]

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

思路

1. 递归函数 以左节点 - 根节点 - 右节点顺序遍历
2. 调用stack
   1. 左边有节点的话 一直放到stack中 一直向左
   2. 取出stack的头节点tmp，放入答案数组
   3. tmp = tmp->right 到右子树

代码

#include <stdio.h>
#include <iostream>
#include <vector>
#include <stack>
using namespace std;
struct TreeNode {
    int val;
    TreeNode *left;
    TreeNode *right;
    TreeNode(int x) : val(x), left(NULL), right(NULL) {}
};
/*
 递归
class Solution {
public:
    vector<int> inorderTraversal(TreeNode* root) {
        vector<int> ans;
        dfs(ans,root);
        return ans;
    }
    void dfs(vector<int>& ans,TreeNode* root){
        if(root==nullptr){return;}
        dfs(ans,root->left);
        ans.push_back(root->val);
        dfs(ans,root->right);
    }
};
 */
/*
 先序遍历
vector<int> preorderTraversal(TreeNode* root) {
    stack<TreeNode*> S;
    vector<int> v;
    TreeNode* rt = root;
    while(rt || S.size()){
        while(rt){
            v.push_back(rt->val);
            S.push(rt->right);
            rt=rt->left;
        }
        rt=S.top();S.pop();
    }
    return v;
}
 
后序遍历
vector<int> preorderTraversal(TreeNode* root) {
    stack<TreeNode*> S;
    vector<int> v;
    TreeNode* rt = root;
    while(rt || S.size()){
        while(rt){
            v.push_back(rt->val);
            S.push(rt->left);
            rt=rt->right;
        }
        rt=S.top();S.pop();
    }
    reverse(v.begin(),v.end());
    return v;
}
*/
//中序
class Solution {
public:
    vector<int> inorderTraversal(TreeNode* root) {
        stack<TreeNode*> sta;
        vector<int> ans;
        TreeNode* tmp = root;
        while (tmp || sta.size()) {
            while(tmp){
                sta.push(tmp);
                tmp = tmp->left;
            }
            tmp = sta.top();sta.pop();
            ans.push_back(tmp->val);
            tmp = tmp->right;
        }
        return ans;
    }
};
int main()
{
    TreeNode root(1), node1(2),node2(3);
    root.right = &node1;
    node1.left = &node2;
    Solution s;
    vector<int> ans = s.inorderTraversal(&root);
    for(int x:ans)
        cout<<x<<endl;
    return 0;
}

今天也是爱zz的一天哦！