二叉树的下一个结点
题目
给定一个二叉树和其中的一个结点,请找出中序遍历顺序的下一个结点并且返回。注意,树中的结点不仅包含左右子结点,同时包含指向父结点的指针。
思路
我们以上图为例进行讲解,上图二叉树的中序遍历是d,b,h,e,i,a,f,c,g。我们以这棵树为例来分析如何找出二叉树的下一个结点。
1>如果结点是它父结点的左子结点,那么它的下一个结点就是它的父结点。例如,途中结点d的下一个结点是b,f的下一个结点是c。
2>如果一个结点既没有右子树,并且它还是父结点的右子结点,这种情形就比较复杂。我们可以沿着指向父结点的指针一直向上遍历,直到找到一个是它父结点的左子结点的结点。
如果这样的结点存 在,那么这个结点的父结点就是我们要找的下一个结点。例如,为了找到结点g的下一个结点,我们沿着指向父结点的指针向上遍历,先到达结点c。由于结点c是
父结点a的右结点,我们继续向上遍历到达结点a。由于结点a是树的根结点。它没有父结点。因此结点g没有下一个结点。
#include <iostream> using namespace std;struct node {int val;struct node *l,*r,*p; }; class BTree {public:BTree();void create(struct node *&r);void add_parent(struct node *&r,struct node *p);node *get_next(struct node *r);struct node *root; }; BTree::BTree() {root=NULL; } void BTree::create(struct node *&r) {int x;cin>>x;if(x==0)return;else{r=new node();r->val=x;create(r->l);create(r->r);} } void BTree::add_parent(struct node *&r,struct node *p) {if(r==nullptr)return;r->p=p;p=r;add_parent(r->l,p);add_parent(r->r,p); } node *BTree::get_next(struct node *r) {if(r==nullptr)return nullptr;if(r->r!=nullptr){struct node *curr=r->r;while(curr->l!=nullptr)curr=curr->l;return curr;}else if(r->p!=nullptr){struct node *curr=r;struct node *p=curr->p;while(p!=nullptr&&curr==p->r){curr=p;p=p->p;}return curr;}return nullptr; } int main() {BTree b;b.create(b.root);b.add_parent(b.root,nullptr);struct node *t=b.get_next(b.root);//测试 if(t!=nullptr)cout<<t->val<<endl;elsecout<<" 该结点无下一个结点."<<endl;return 0; }
转载于:https://www.cnblogs.com/tianzeng/p/10129024.html
总结
- 上一篇: 钠离子电池能量转化形式
- 下一篇: Git 工作流程