Leetcode 刷題總結
阿新 • • 發佈:2022-03-16
- 二叉樹的中序遍歷(非遞迴版)
點選檢視程式碼
/** * 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) { vector<int> res; inOrder(root, res); return res; } void inOrder(TreeNode* root, vector<int>& res){ // 使用棧 stack<TreeNode*> st; TreeNode *p = root; while(p || !st.empty()){ if(p){ st.push(p); p = p->left; } else{ TreeNode *node = st.top(); st.pop(); res.push_back(node->val); p = node->right; } } } };