94.Binary Tree Inorder Traversal

問題概述

Given a binary tree, return the inorder traversal of its nodes’ values.

Example

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

Output: [1,3,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) {
        vector<int> vector;
        if(!root)
        return vector;
        stack<TreeNode *> stack;
        stack.push(root);
        while(!stack.empty())
        {
            TreeNode *pNode = stack.top();
            if(pNode->left)
            {
                stack.push(pNode->left);
                pNode->left = NULL;
            }
            else
            {
                vector.push_back(pNode->val);
                stack.pop();
                if(pNode->right)
                stack.push(pNode->right);
            }
        }
        return vector;
    }
};