面试题36：二叉搜索树与双向链表

83 阅读 0 评论 55 点赞

我是靠谱客的博主帅气石头，这篇文章主要介绍面试题36：二叉搜索树与双向链表，现在分享给大家，希望可以做个参考。

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

复制代码

/*
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)
    {
        if(pRootOfTree == nullptr)
            return nullptr;
        TreeNode* pLast = nullptr;//指向双向链表的尾节点
        convertNode(pRootOfTree, &pLast);//将子树转为链表
        TreeNode* pHead = pLast;
        while(pHead != nullptr && pHead->left != nullptr)
            pHead = pHead->left;//沿双向链表向前遍历找到头节点
        return pHead;
    }

void convertNode(TreeNode* pRoot, TreeNode** pLast)
    {
        if(pRoot == nullptr)
            return;
        TreeNode* pCur = pRoot;
        if(pCur->left != nullptr)
            convertNode(pCur->left, pLast);
        pCur->left = *pLast;        //将根节点插入左子树形成的
        if(*pLast != nullptr)       //链表的 末端
            (*pLast)->right = pCur; //
        *pLast = pCur;//更新当前链表的尾节点
        if(pCur->right != nullptr)
            convertNode(pCur->right,pLast);
    }
};

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
/*
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)
    {
        if(pRootOfTree == nullptr)
            return nullptr;
        TreeNode* pLast = nullptr;//指向双向链表的尾节点
        convertNode(pRootOfTree, &pLast);//将子树转为链表
        TreeNode* pHead = pLast;
        while(pHead != nullptr && pHead->left != nullptr)
            pHead = pHead->left;//沿双向链表向前遍历找到头节点
        return pHead;
    }

    void convertNode(TreeNode* pRoot, TreeNode** pLast)
    {
        if(pRoot == nullptr)
            return;
        TreeNode* pCur = pRoot;
        if(pCur->left != nullptr)
            convertNode(pCur->left, pLast);
        pCur->left = *pLast;        //将根节点插入左子树形成的
        if(*pLast != nullptr)       //链表的 末端
            (*pLast)->right = pCur; //
        *pLast = pCur;//更新当前链表的尾节点
        if(pCur->right != nullptr)
            convertNode(pCur->right,pLast);
    }
};