LeetCode 94.二叉树的中序遍历 C++

题目描述

定一个二叉树的根节点 root ，返回它的 中序 遍历。

示例 1：
输入：root = [1,null,2,3]
输出：[1,3,2]

示例 2：

输入：root = []
输出：[]

示例 3：

输入：root = [1]
输出：[1]

示例 4：
输入：root = [1,2]
输出：[2,1]

示例 5：
输入：root = [1,null,2]
输出：[1,2]
 

提示：

树中节点数目在范围 [0, 100] 内
-100 <= Node.val <= 100
 

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

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/binary-tree-inorder-traversal

解题思路1

用dfs的思想，中序遍历套路
中序遍历一下树，并且在访问该结点的时候把该结点存到数组中

代码1

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode() : val(0), left(nullptr), right(nullptr) {}
 *     TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
 *     TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {}
 * };
 */
class Solution {
public:
    void inOrder(TreeNode* root,vector<int> &re){
        if(root==nullptr){  //遇到空结点，返回
            return;
        }
        inOrder(root->left,re); //中序遍历左子树
        re.push_back(root->val);    //把访问的结点存入数组
        inOrder(root->right,re);    //中序遍历右子树
    }
    vector<int> inorderTraversal(TreeNode* root) {
        vector<int> res;    //定义数组用来存储最后的结果
        inOrder(root, res); //中序遍历该树
        return res; //返回结果
    }
};

解题思路2

用栈模拟递归调用
在结点非空或者栈非空的情况下：
1.如果该结点非空，那么入栈，继续遍历左子树。
2.如果当前结点为空结点，回溯到上一个结点，并用root记录。存入数组，出栈，并且访问root的右孩子。
3.最后结点为空并且栈也为空，说明访问结束了。返回结果res。

代码2

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode() : val(0), left(nullptr), right(nullptr) {}
 *     TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
 *     TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {}
 * };
 */
class Solution {
public:
    vector<int> inorderTraversal(TreeNode* root) {
        stack<TreeNode*> stk;   //栈用来模拟递归的过程
        vector<int> res;    //用来记录结点值的数组
        if(root==NULL){ //特判，如果跟结点为空，直接返回空数组，其实这行可以不用写。
            return res;
        }
        while(root!=NULL||!stk.empty()){    //结点不空或者栈不空时
            if(root!=NULL){ //如果当前结点为不空
                stk.push(root); //入栈
                root=root->left;    //访问其左孩子
            }else{  //当前结点为空的情况
                root=stk.top(); //root指向栈顶元素，即回溯的过程
                res.push_back(root->val);   //将root的值存入数组
                stk.pop();  //出栈
                root=root->right;//指向其右孩子
            }
        }
        return res;//最终返回结果
    }
};