二叉树的中序遍历——迭代算法

题目

二叉树的中序遍历——迭代算法_第1张图片

思路

中序遍历的思想是:左子树——根节点——右子树
所以相应的思路为:

  1. 不断寻找左子树根节点
  2. 当该左子树为空时,输出该节点
  3. 遍历右子树

相应代码如下:

class Solution {
public:
    vector inorderTraversal(TreeNode* root) {
        vector v;
		stack s;
		TreeNode* tre = root;
		while (tre || !s.empty()) {
			while (tre) {//不断寻找左节点
				s.push(tre);
				tre = tre->left;
			}
			//当左子树为空时,则将该节点从栈中弹出
			tre = s.top();
			s.pop();
			v.push_back(tre->val);
			tre = tre->right;//最后遍历右子树
		}
		return v;
    }
};

你可能感兴趣的:(LeetCode)