【整理】把二元查找树转变成排序的双向链表

整理自:

1. July @CSDN http://blog.csdn.net/v_JULY_v/article/details/6278484

2.剑指offer by 何海涛 http://zhedahht.blog.163.com/blog/static/254111742007127104759245/

修改 by sail2011@新浪微薄:http://weibo.com/sail2011  

 

题目:输入一棵二元查找树,将该二元查找树转换成一个排序的双向链表。要求不能创建任何新的结点,只调整指针的指向。

  比如将二元查找树
                                         10
                                        /    \
                                      6       14
                                    /  \     /  \
                                 4     8  12   16
转换成双向链表

4=6=8=10=12=14=16。

  分析:本题是微软的面试题。很多与树相关的题目都是用递归的思路来解决,本题也不例外。下面我们用两种不同的递归思路来分析。

何海涛:

      在二叉树中,每个结点都有两个指向子结点的指针。在双向链表中,每个结点也有两个指针,它们分别指向前一个结点和后一个结点。由于这两种结点的结构相似,同时二叉搜索树也是一种排序的数据结构,因此在理论上有可能实现二叉搜索树和排序的双向链表的转换。在搜索二叉树中,左子结点的值总是小于父结点的值,右子结点的值总是大于父结点的值。因此我们在转换成排序双向链表时,原先指向左子结点的指针调整为链表中指向前一个结点的指针,原先指向右子结点的指针调整为链表中指向后一个结点指针。接下来我们考虑该如何转换。

      由于要求转换之后的链表是排好序的,我们可以中序遍历树中的每一个结点,这是因为中序遍历算法的特点是按照从小到大的顺序遍历二叉树的每一个结点。当遍历到根结点的时候,我们把树看成三部分:值为10的结点、根结点值为6的左子树、根结点值为14的右子树。根据排序链表的定义,值为10的结点将和它的左子树的最大一个结点(即值为8的结点)链接起来,同时它还将和右子树最小的结点(即值为12的结点)链接起来。

      按照中序遍历的顺序,当我们遍历转换到根结点(值为10的结点)时,它的左子树已经转换成一个排序的链表,并且处在链表中的最后一个结点是当前值最大的结点。我们把值为8的结点和根结点链接起来,此时链表中的最后一个结点就是10了。接着我们去遍历转换右子树,并把根结点和右子树中最小的结点链接起来。至于怎么去转换它的左子树和右子树,由于遍历和转换过程是一样的,我们很自然地想到可以用递归。

本题考点和分析:

      考查应聘者分析复杂问题的能力。无论是二叉树还是双向链表,都有很多指针。要实现这两种不同数据结构的转换,需要调整大量的指针,因此这个过程会很复杂。为了把这个复杂的问题分析清楚,我们可以把树分为三个部分:根结点、左子树和右子树,然后把左子树中最大的结点、根结点、右子树中最小的结点链接起来。至于如何把左子树和右子树内部的结点链接成链表,那和原来的问题的实质是一样的,因此可以递归解决。解决这个问题的关键在于把一个大的问题分解成几个小问题,并递归地解决小问题。

  思路一:当我们到达某一结点准备调整以该结点为根结点的子树时,先调整其左子树将左子树转换成一个排好序的左子链表,再调整其右子树转换右子链表。最近链接左子链表的最右结点(左子树的最大结点)、当前结点和右子链表的最左结点(右子树的最小结点)。从树的根结点开始递归调整所有结点。

  思路二:我们可以中序遍历整棵树。按照这个方式遍历树,比较小的结点先访问。如果我们每访问一个结点,假设之前访问过的结点已经调整成一个排序双向链表,我们再把调整当前结点的指针将其链接到链表的末尾。当所有结点都访问过之后,整棵树也就转换成一个排序双向链表了。

参考代码:

首先我们定义二元查找树结点的数据结构如下:

structBSTreeNode // a node in the binary search tree
    {
        int         m_nValue; // value of node
        BSTreeNode  *m_pLeft;  // left child of node
        BSTreeNode  *m_pRight; // right child of node
    };

何海涛解法一:

///
// Covert a sub binary-search-tree into a sorted double-linked list
// Input: pNode - the head of the sub tree
//        asRight - whether pNode is the right child of its parent
// Output: if asRight is true, return the least node in the sub-tree
//         else return the greatest node in the sub-tree
///
BSTreeNode* ConvertNode(BSTreeNode* pNode, boolasRight)
{
      if(!pNode)
            returnNULL;
                        
      BSTreeNode *pLeft = NULL;
      BSTreeNode *pRight = NULL;
                        
      // Convert the left sub-tree
      if(pNode->m_pLeft)
            pLeft = ConvertNode(pNode->m_pLeft, false);
                        
      // Connect the greatest node in the left sub-tree to the current node
      if(pLeft)
      {
            pLeft->m_pRight = pNode;
            pNode->m_pLeft = pLeft;
      }
                        
      // Convert the right sub-tree
      if(pNode->m_pRight)
            pRight = ConvertNode(pNode->m_pRight, true);
                        
      // Connect the least node in the right sub-tree to the current node
      if(pRight)
      {
            pNode->m_pRight = pRight;
            pRight->m_pLeft = pNode;
      }
                        
      BSTreeNode *pTemp = pNode;
                        
      // If the current node is the right child of its parent, 
      // return the least node in the tree whose root is the current node
      if(asRight)
      {
            while(pTemp->m_pLeft)
                  pTemp = pTemp->m_pLeft;
      }
      // If the current node is the left child of its parent, 
      // return the greatest node in the tree whose root is the current node
      else
      {
            while(pTemp->m_pRight)
                  pTemp = pTemp->m_pRight;
      }
                         
      returnpTemp;
}
                        
///
// Covert a binary search tree into a sorted double-linked list
// Input: the head of tree
// Output: the head of sorted double-linked list
///
BSTreeNode* Convert(BSTreeNode* pHeadOfTree)
{
      // As we want to return the head of the sorted double-linked list,
      // we set the second parameter to be true
      returnConvertNode(pHeadOfTree, true);
}

何海涛解法二(推荐):

// Covert a sub binary-search-tree into a sorted double-linked list
// Input: pNode -           the head of the sub tree
//        pLastNodeInList - the tail of the double-linked list
void ConvertNode(BSTreeNode* pNode, BSTreeNode*& pLastNodeInList)
{
      if(pNode == NULL)
            return;
                                                
      BSTreeNode *pCurrent = pNode;
                                                
      // Convert the left sub-tree
      if (pCurrent->m_pLeft != NULL)
            ConvertNode(pCurrent->m_pLeft, pLastNodeInList);
                                                
      // Put the current node into the double-linked list
      pCurrent->m_pLeft = pLastNodeInList;
      if(pLastNodeInList != NULL)
            pLastNodeInList->m_pRight = pCurrent;
                                                
      pLastNodeInList = pCurrent;
                                                
      // Convert the right sub-tree
      if (pCurrent->m_pRight != NULL)
            ConvertNode(pCurrent->m_pRight, pLastNodeInList);
}
                                                
// Covert a binary search tree into a sorted double-linked list
// Input: pHeadOfTree - the head of tree
// Output: the head of sorted double-linked list
BSTreeNode* Convert_Solution1(BSTreeNode* pHeadOfTree)
{
      BSTreeNode *pLastNodeInList = NULL;
      ConvertNode(pHeadOfTree, pLastNodeInList);
                                                
      // Get the head of the double-linked list
      BSTreeNode *pHeadOfList = pLastNodeInList;
      while(pHeadOfList && pHeadOfList->m_pLeft)
            pHeadOfList = pHeadOfList->m_pLeft;
                                                
      return pHeadOfList;
}


sail2011 注:

      先从宏观上来考虑问题,把中序遍历和递归的流程想清楚! 对于july的第一个程序,在每一次ergodicBSTree(pCurrent->m_pLeft)执行后都要返回一次,也就是第一次pCurrent是1号节点,然后pCurrent是2号节点,再然后是4号节点。。。。。。

    4

 /     \

2         6

/   \     /

1   3   5

 

july程序:

// 遍历二元查找树  中序  
voidergodicBSTree(BSTreeNode * pCurrent)  
{ 
    if(NULL == pCurrent)  
    {        
        return; 
    } 
    if(NULL != pCurrent->m_pLeft)  
    { 
        ergodicBSTree(pCurrent->m_pLeft);    
    } 
                                                      
    // 节点接到链表尾部  
    convertToDoubleList(pCurrent); 
    // 右子树为空  
    if(NULL != pCurrent->m_pRight)  
    { 
        ergodicBSTree(pCurrent->m_pRight); 
    } 
} 
                                                     
// 二叉树转换成list  
void convertToDoubleList(BSTreeNode * pCurrent)  
{ 
                                                      
    pCurrent->m_pLeft = pListIndex;  
    if(NULL != pListIndex)  
    { 
        pListIndex->m_pRight = pCurrent;  
    } 
    else
    { 
        pHead = pCurrent;  
    }    
    pListIndex = pCurrent;  
    cout<<pCurrent->m_nValue<<endl; 
}


july: 但显然,以下这种思路更容易理解些:(注释by sail2011)

BSTreeNode* ConvertNode(BSTreeNode* pNode, bool asRight)
{ 
      if(!pNode) 
            return NULL; 
                                                        
      BSTreeNode *pLeft = NULL;   //用来存储当前结点的左子树的最后一个结点
      BSTreeNode *pRight = NULL;  //用来存储当前结点的右子树的第一个结点
                                                        
      // 如果当前结点有左子树就转换其左子树
      if(pNode->m_pLeft) 
            //存储当前结点的左子树的最后一个结点
        pLeft = ConvertNode(pNode->m_pLeft, false); 
                                                        
      // 连接左子树中最大的结点即pLeft到当前结点
      if(pLeft) 
      { 
            pLeft->m_pRight = pNode; 
            pNode->m_pLeft = pLeft; 
      } 
                                                        
      // 如果当前结点有右子树就转换其右子树
      if(pNode->m_pRight) 
            //存储当前结点的右子树的第一个结点
            pRight = ConvertNode(pNode->m_pRight, true); 
                                                        
      // 连接右子树中最小的结点即pRight到当前结点
      if(pRight) 
      { 
            pNode->m_pRight = pRight; 
            pRight->m_pLeft = pNode; 
      } 
                                                        
      BSTreeNode *pTemp = pNode; 
                                                        
      // 如果当前结点是某个右子树就返回该右子树中最小的结点
      if(asRight) 
      { 
            while(pTemp->m_pLeft) 
                  pTemp = pTemp->m_pLeft; 
      } 
      // 如果当前结点是某个左子树就返回该左子树中最大的结点
      else
      { 
            while(pTemp->m_pRight) 
                  pTemp = pTemp->m_pRight; 
      } 
                                                      
      //返回结点地址给ConvertNode
      //左子树为最后一个结点,右子树为第一个结点
      return pTemp; 
}


整理自:1. July @CSDNhttp://blog.csdn.net/v_JULY_v/article/details/6278484

2.剑指offer by 何海涛  http://zhedahht.blog.163.com/blog/static/254111742007127104759245/

修改by sail2011@新浪微薄:http://weibo.com/sail2011  

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值