给定一个二叉树的根节点 root
,返回 它的 中序 遍历 。
中序遍历:按照左子树-根节点-右子树的顺序来遍历二叉树。
迭代的思路是利用了栈的先进后出的特点进行遍历的,具体的代码实现见下面
看不懂代码的可以先思考下面两个问题
/**
* 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> vec;
stack<TreeNode*> s;
while(root != nullptr || !s.empty() ){
while(root != nullptr){
s.push(root);
root = root->left;
}
root = s.top();
vec.push_back(root->val);
root = root ->right;
s.pop();
}
return vec;
}
};
1.为什么用两个while循环:因为中序遍历是按照左根右的顺序来进行遍历的,所以需要一个while循环来先遍历到二叉树的最左边节点即为第二个while的作用。
2.为什么第一个while循环条件不一样:其实这样迭代的思路可以这样理解,即先找到二叉树的最左边节点后下一步还会执行root = root->left,这样其实就是说最左边节点的左边节点为空,而最左边的节点反而可以作为其根节点,我们只需要让指针再次指向其右边节点就完成了左根右的遍历,但由于其为最左边节点,必定没有右子树和他相连,故root = root->right后为空,如果第一个循环while不加条件的话就不能执行下去。