递归写法
/** * 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> res; void dfs(TreeNode* root) { if(!root) return; dfs(root->left); res.push_back(root->val); dfs(root->right); } vector<int> inorderTraversal(TreeNode* root) { dfs(root); return res; } }; 作者:Once. 链接:https://www.acwing.com/activity/content/code/content/3201627/ 来源:AcWing 著作权归作者所有。商业转载请联系作者获得授权,非商业转载请注明出处。
非递归写法(利用栈,固定套路,要求背会)
/** * 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; // 定义答案数组 stack<TreeNode*> stk; // 定义栈 while(root || stk.size()) // 当当前结点非空或栈非空时 { while(root) // 当当前结点不空时 { stk.push(root); // 将当前结点入栈 root = root->left; // 当前结点走到其左儿子的位置 } root = stk.top(); // 将栈顶元素的值取出 stk.pop(); // 将栈顶元素弹出 res.push_back(root->val); // 遍历当前点 root = root->right; // 遍历完之后走到当前结点的右儿子 } return res; // 返回答案数组 } }; 作者:Once. 链接:https://www.acwing.com/activity/content/code/content/3201627/ 来源:AcWing 著作权归作者所有。商业转载请联系作者获得授权,非商业转载请注明出处。