leetcode原题链接:二叉树的中序遍历
题目描述
给定一个二叉树的根节点 root
,返回 它的 中序 遍历 。
示例 1:
输入:root = [1,null,2,3] 输出:[1,3,2]
示例 2:
输入:root = [] 输出:[]
示例 3:
输入:root = [1] 输出:[1]
提示:
- 树中节点数目在范围
[0, 100]
内 -100 <= Node.val <= 100
解题方法:用一个stack依次将root的一层层的left节点压入栈中,所有的left节点压入栈后,栈顶元素则为当前最左侧的节点,则访问该节点,访问完后将该节点的right节点压入栈中,并重复上述过程直到栈顶为空。
C++代码
#include <iostream>
#include <stack>
#include <vector>
/**
* 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) {
if (!root) {
return {};
}
TreeNode* p = root;
std::stack<TreeNode*> s;
std::vector<int> result;
while(p || !s.empty()) {
while(p) {
s.emplace(p);
p = p->left;
}
TreeNode* p_top = s.top();
s.pop();
if (!p_top) { //leetcode对所有的指针都有非空的判断,否则编译无法通过
continue;
}
result.emplace_back(p_top->val);
if (p_top->right) {
p = p_top->right;
}
}
return result;
}
};