二叉树中序遍历方法实现
对于二叉树的遍历,先序的方式是比较简单的,但是中序和后序的方式还是有点麻烦的,这里先给出一个用C++stack的遍历方式:
1.如果当前结点不为空 把当前结点压入栈 p=p->left转向其左孩子 2.如果当前结点为空(证明这半棵子树已经遍历完成,需要从栈顶找到树根) 取栈顶元素为当前结点,栈做一次弹栈操作 访问当前结点 p=p->right转向其右孩子 3.重复1、2操作,直到栈为空并且当前结点为NULL,结束。
算法代码实现:
vector<int> inOrder(TreeNode *root) { vector<int> result; const TreeNode *cur; stack<const TreeNode*> s; cur = root; while(!s.empty() || cur != NULL) { if(cur != NULL) { s.push(cur); cur = cur->left; }//if else { cur = s.top(); s.pop(); result.push_back(cur->val); cur = cur->right; } }//while return result; }
这种利用栈的遍历方法很简单,在之前我也设计了一个自己的遍历的方法:
1.如果当前结点为空(NULL) 取栈顶元素为当前结点,栈做一次弹栈操作 访问当前结点 p=p->right转向其右孩子 2.如果当前结点不为空 如果当前结点有左孩子 把当前结点压入栈 p=p->left转向其左孩子 如果当前结点没有左孩子 访问当前结点 p=p->right转向其右孩子 3.重复1、2操作,直到栈为空并且当前结点为NULL,结束。
代码实现:
vector<int> inOrder(TreeNode *root) { const TreeNode* cur; stack<const TreeNode*> s; vector<int> result; cur = root; while(!s.empty() || cur != NULL) { if(cur == NULL) { cur = s.top(); s.pop(); result.push_back(cur->val); cur = cur->right; }//if else { if(cur->left != NULL) { s.push(cur); cur = cur->left; } else { result.push_back(cur->val); cur = cur->right; } }//else }//while return result; }
上面介绍了两种利用栈对树中序遍历的方法,下面还是说一下Morris的遍历方法
1.如果当前结点有左子树 找到左子树的最右叶子结点 如果最右叶子结点的右指针域为空(NULL) 使最右叶子结点的右指针域指向当前结点 p=p->left转向其左子树 如果最右叶子结点的右指针域已经指向了当前结点 访问当前结点 断开Morris链接 p=p->right转向其子树 2.如果当前结点没有左子树 访问当前结点 p=p->right转向其子树 3.重复1、2操作,直到当前结点为NULL,结束。
算法代码实现:
vector<int> InOrder(TreeNode *root) { TreeNode *cur = NULL; TreeNode *temp = NULL; cur = root; vector<int> result; while(cur != NULL) { if(cur->left != NULL) { temp = cur->left; while(temp->right != NULL && temp->right != cur) { temp = temp->right; } if(temp->right == NULL) { temp->right = cur; cur = cur->left; } else //temp->right == cur { result.push_back(cur->val); temp->right = NULL; cur = cur->right; } } else //没有左子树 { result.push_back(cur->val); cur = cur->right; } }//while return result; }