剑指offer-二叉搜索树与双向链表

玩转PPT 2018-04-14

题目描述

输入一棵二叉搜索树,将该二叉搜索树转换成一个排序的双向链表。要求不能创建任何新的结点,只能调整树中结点指针的指向。

解题思路

问题是将二叉搜索树转换成一个排序的双向链表,考虑到二叉搜索树的中序遍历序列为排序序列,所以可用中序遍历的递归做法来调整指针。在遍历过程中,调整指针pLast使其指向当前已转换好双向链表的末尾也就是最大数字,这样每当遍历到一个节点,只需要调整两次指针:

  1. 调整当前结点的left指针指向pLast
  2. 调整pLast指向结点的right指针指向当前结点

这样递归调整完成后,pLast指向的就是双向链表的最后一个节点,要返回的是头结点,所以依次向前调整pLast直至其左指针为空即指向头结点。

代码

/*
 struct TreeNode {
     int val;
     struct TreeNode *left;
     struct TreeNode *right;
     TreeNode(int x) :
             val(x), left(NULL), right(NULL) {
     }
 };*/
 class Solution {
 public:
     TreeNode* pLast=NULL;
     TreeNode* Convert(TreeNode* pRoot)
     {
         if(!pRoot)
             return NULL;
         ToList(pRoot);
         while(pLast->left)
             pLast=pLast->left;
         return pLast;
     }
     void ToList(TreeNode* pRoot){
         if(pRoot){
             if(pRoot->left)
                 ToList(pRoot->left);
             if(pLast){
                 pLast->right=pRoot;
                 pRoot->left=pLast;
             }
             pLast=pRoot;
             if(pRoot->right)
                 ToList(pRoot->right);
         }
     }
 };

相关推荐