js找出中序遍历的下一个子节点

题目:

给定一个二叉树和其中的一个结点,请找出中序遍历顺序的下一个结点并且返回。注意,树中的结点不仅包含左右子结点,同时包含指向父结点的指针。

思路:

下图中序遍历为:DBFEGAC

  • 第一种情况:如果给定节点右子树不为空,那么该节点的下一个节点是右子树的最左节点
    例如:给定B,则下一个节点为F在这里插入图片描述
  • 否则: 第二种情况:没有右子树时,那么就得找父节点,当父节点的左等于当前节点也就找到了
1.当前节点可能为上图的D,则下一个节点为B,return B
2.当前节点可能为上图的G,则下一节点为A,return A
3.当前节点可能为上图的C,则下一节点为nul,因为已经遍历完了  上图中序遍历为:DBFEGAC

js代码

function GetNext(pNode) {
	  if (!pNode) {
       		 return null;
      }
		//如果当前结点右子树不空,那么中序下一个节点就是右子树的最左节点;如果右子树没有左子树就直接返回右子树根节点
	  if (pNode.right!=null) {
			pNode=pNode.right;
			while(pNode.left) { //一直找到最下左子树
				pNode=pNode.left;
			}
			return pNode;
		}
		
		//右节点为空的两种情况
		//1、
		//如果 父节点的右节点 等于 当前的节点,继续返回到上层的父节点...直到   父节点左子节点  等于  当前节点 
		while (pNode.next!=null&&pNode.next.right==pNode) {
			pNode=pNode.next;
		}
		//2、
		//如果右子树为空,返回上层父节点
		return pNode.next;
		
    }
 }

猜你喜欢

转载自blog.csdn.net/HZ___ZH/article/details/110861156