【Leetcode】NO.94 二叉树的中序遍历 (C++&Python) [二叉树]

题目：94. 二叉树的中序遍历

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

思路

• 迭代法使用中序遍历

1. 一直遍历到做子树的最底层
2. 再pop出当前节点，之后再遍历右子树

开辟一个栈存节点，一个数组保存节点的值
判断节点是否为空，或者栈是否为空

代码

C++ 版本

/**
 * 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*> s;
        vector<int> result;
        TreeNode *cur = root;
        while(cur!=NULL || !s.empty())
        {

         if(cur!=NULL) //将一直遍历到最左边
         {
             s.push(cur);
             cur = cur->left;
         }   
         else
         {
             // 把点Put
             cur = s.top();
             s.pop(); //将数据弹出了
             result.push_back(cur->val);
             cur = cur->right;
         }

        }
        return result;
    }
};

Python 版本

# Definition for a binary tree node.
# class TreeNode:
#     def __init__(self, val=0, left=None, right=None):
#         self.val = val
#         self.left = left
#         self.right = right
class Solution:
    def inorderTraversal(self, root: Optional[TreeNode]) -> List[int]:
        result = [] # 保存节点
        s = []
        cur = root
        while cur or s:
        
            if cur:
              s.append(cur)
              cur = cur.left;
            else:
                cur = s.pop()
                result.append(cur.val)
                cur = cur.right
        
        return result