Leetcode 094 二叉树的中序遍历

题目描述：

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

示例:

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

输出: [1,3,2]

 

思路：使用栈来完成

1.先将根节点及其所有左子节点放入栈

2.取出栈顶元素，保存节点值，若其有右子节点，则在下次循环时将此右子节点的左子节点压入栈

 

代码：

/**
 * Definition for a binary tree node.
 * 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) {
        
        stack<TreeNode*> temp;
        vector<int> test;
        TreeNode* p = root;
        
        while(temp.size() != 0 || p)
        {
            //将所有左子节点压入栈
            while(p)
            {
                temp.push(p);
                p = p -> left;
            }
            //取出栈顶元素，若此节点有右子节点,则下次循环时能将其左子节点压入栈
            p = temp.top();
            test.push_back(p -> val);
            temp.pop();
            
            p = p -> right;
        }
        
        return test;
    }
};