题意:给一个二叉树(树结构),返回中序遍历(左子节点,根节点,右子节点)。
solution 1: 递归无需多言,但是需要逐层调用,而且压栈了冗余的信息。效率低。
solution 2: 了解中序遍历的原理,手动模拟压栈的过程。如果有左子节点,不断地将左子节点压栈;没有左子节点了,那么此时应该访问其根节点,因此出栈栈顶元素,此时应该访问右子节点,因此将右子节点压栈。循环,直到空栈且当前节点为null。
note: 像这种节点可能是null的链式结构,不需要判断后续节点是否是null,只需要判断当前节点即可。只需要在访问当前节点之前判断是否是null即可。
/**
* Definition for a binary tree node.
* struct TreeNode {
* int val;
* TreeNode *left;
* TreeNode *right;
* TreeNode(int x) : val(x), left(NULL), right(NULL) {}
* };
*/
class Solution {
public:
vector<int> inorderTraversal(TreeNode* root) {
//1. recursivie
/*vector<int> ret;
if ( root == NULL )
return ret;
vector<int> left, right;
left = inorderTraversal(root->left);
ret.insert(ret.end(), left.begin(),left.end());
ret.push_back(root->val);
right = inorderTraversal(root->right);
ret.insert(ret.end(), right.begin(), right.end());
return ret;*/
//2.stack: 如果有左子节点,不断地将左子节点压栈;没有左子节点了,那么将该节点记录(出栈),右子节点压栈。循环,直到空栈
vector<int> ret;
stack<TreeNode*> s;
TreeNode* s_top = root;
while ( s_top!=NULL || !s.empty() ) {
while ( s_top != NULL ) {
s.push(s_top);
s_top = s_top->left;
}
s_top = s.top();
s.pop(); // !!!返回none
ret.push_back( s_top->val );
s_top = s_top->right;
}
return ret;
}
};
solution 3: morris traversal
堆栈解的submission结果:
使用堆栈排名依然很差。看来morris traversal的方法被很多人采纳了。以后有时间值得学习。