这道题的核心并不是让你去用递归实现二叉树的中序遍历,而是考察使用多种遍历的方法去实现,包括使用栈遍历,计数法,莫里斯遍历等。
首先就是计数法,计数法的意思就是根据当前节点被访问的次数去判断是否要输出当前节点的值。在中序遍历中,当出现第二次的时候就可以将节点的值输出了,同理该方法也可以推广到前序和后序遍历中去,只要修改一下当前节点的入栈顺序即可,C++代码如下:
class Solution {
public:
vector<int> inorderTraversal(TreeNode* root) {
stack<TreeNode*> node_stack;
map<TreeNode*, int> node_cnt;
vector<int> ans;
TreeNode* now;
if(root == NULL) return ans;
node_stack.push(root);
while(!node_stack.empty()){
now = node_stack.top();
node_stack.pop();
if(now == NULL) continue;
if(node_cnt.find(now) == node_cnt.end()){
node_cnt[now] = 1;
// 这里可以通过修改now节点的入栈顺序
// 改变遍历的顺序
node_stack.push(now->right);
node_stack.push(now);
node_stack.push(now->left);
/* 前序遍历 */
/*
node_stack.push(now->right);
node_stack.push(now->left);
node_stack.push(now);
*/
/* 后序遍历 */
/*
node_stack.push(now);
node_stack.push(now->right);
node_stack.push(now->left);
*/
}
else if(node_cnt[now] == 1){
ans.push_back(now->val);
}
}
return ans;
}
};
然后就是莫里斯遍历法,莫里斯遍历其实就是线索二叉树的感觉,每次都找到当前节点左子树的最右节点,然后把最右节点的右子树赋值为当前节点,具体代码如下:
class Solution {
public:
vector<int> inorderTraversal(TreeNode* root) {
stack<TreeNode*> node_stack;
vector<int> ans;
TreeNode* now = root;
TreeNode* pre;
if(root == NULL) return ans;
node_stack.push(root);
while(now != NULL){
if(now->left == NULL){
ans.push_back(now->val);
now = now->right;
}
else{
pre = now;
now = now->left;
while(now->right != NULL){
now = now->right;
}
now->right = pre;
now = pre->left;
pre->left = NULL;
}
}
return ans;
}
};