LetCode 94. 二叉樹的中序遍歷

/**
 * 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> res;
//         Traversal(root, res);
//         return res;
//     }
    
//     void Traversal(TreeNode* root, vector<int>& res){
//         if (!root)
//             return;
//         Traversal(root->left, res);
//         res.push_back(root->val);
//         Traversal(root->right, res);
//     }
// };

class Solution {
public:
    vector<int> inorderTraversal(TreeNode* root) {
        vector<int> res;
        stack<TreeNode*> s;
        while(root || !s.empty()){
            // 先將左兒子全部入棧
            while(root){
                // 在這裏訪問則爲先序
                // res.push_back(root->val);
                s.push(root);
                root = root->left;
            }
            // 然後訪問棧頂節點
            if (!s.empty()){
                root = s.top();
                s.pop();
                // 在這裏訪問爲中序
                res.push_back(root->val);
                // 訪問右兒子
                root = root->right;
            }
        }
        return res;
    }
};

static int x=[](){
    std::ios::sync_with_stdio(false);
    cin.tie(NULL);
    return 0;
}();
發佈了304 篇原創文章 · 獲贊 26 · 訪問量 8萬+
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章