刷题笔记：二叉树与双向链表（C++）

剑指offer刷题笔记

 

题目描述

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

举例

 

 

思路

1. 前面有道也是关于的二叉搜索树的笔记提到搜索二叉树的特点
   1. 左子树的数值小于根节点，右子树的数值大于根节点
   2. 二叉搜索树的子树也是二叉搜索树
2. 根据上述的特点和所需要的有序结果，采取中序遍历的方式来遍历该二叉搜索树，便会得到一个有序的双向链表
3. 按中序遍历完之后指针会知道链表中的最后一个结点，而需要返回表头，所以需要找到头结点

 

源代码：

/*
struct TreeNode {
	int val;
	struct TreeNode *left;
	struct TreeNode *right;
	TreeNode(int x) :
			val(x), left(NULL), right(NULL) {
	}
};*/
class Solution {
public:
    TreeNode* Convert(TreeNode* pRootOfTree)
    {
        TreeNode* pLastNodeInList=NULL;
        
        //创建链表
        ConvertNode(pRootOfTree,&pLastNodeInList);
        
        //找头节点
        TreeNode* pHeadOfList=pLastNodeInList;
        while(pHeadOfList!=NULL && pHeadOfList->left!=NULL){
            pHeadOfList=pHeadOfList->left;
        }
        
        return pHeadOfList;
        
    }
    
    void ConvertNode(TreeNode* pNode,TreeNode** pLastNodeInList){
        
        //叶子结点子直接返回
        if(pNode==NULL)
            return;
        
        TreeNode* pCurrent=pNode;
        //递归左子树
        if(pCurrent->left!=NULL)
            ConvertNode(pCurrent->left,pLastNodeInList);
        
        //构建双向链接
        pCurrent->left=*pLastNodeInList;
        if(*pLastNodeInList!=NULL){
            (*pLastNodeInList)->right=pCurrent;
        }
        *pLastNodeInList=pCurrent;
        
        //递归右子树
        if(pCurrent->right!=NULL){
            ConvertNode(pCurrent->right,pLastNodeInList);
        }
    }
};