剑指offer36:二叉搜索树与双向链表

 

一道很有意思的数据结构题

分析:

看到二叉搜索树想到中序遍历（dfs）可以得到一个从小到大的排序序列，题目中说明不能创建新的空间只能修改节点的指向，emmm。

首先对树，又是一棵二叉搜索树，我们很容易想到要用递归去做，而且，先把左子树变成双向链表，再接上当前root，再把右子树也变成双向链表接上，但是就是这么简单的东西，也有挺多麻烦。

1.返回头节点如何返回？如何记录下头节点

2.如何对每个节点实施相同的操作使其能够递归？

对第一个问题，我们可以用一个全局的头节点保存遍历的每一个节点直到最左侧，这很直观，但是我们初始化这个节点的条件是什么，是他的左右子树都为空吗？显然不行，还有其他的叶子节点，或许你还会说用标志变量，记录第一个左子节点为空的点，好吧也不是不行，但是有更好的方法。

对第二个问题，要实现递归，我们最好想好对每个节点实施相同的操作。试想一下，我们到了当前的一个递归层，或许会拥有当前层的一个节点，但是仅有一个节点可以操作么？例如上图中我们到了2这个节点，你或许会想，让其左子树的右指针指向本身就好了，但是到了3怎么办呢，无法得到2的节点无法操作。而想一想我们链表中的操作，我们想到要设置一个前驱指针，记录遍历的前面一个节点，那么中序遍历的前一个节点是不是要小于当前节点的？所以对当前节点的操作变为将前驱节点的右指针指向当前节点，当前节点的左指针指向前驱。

而第一个问题也解决了，如果想找到头节点，在前驱指针初始化的前一点点时刻，赋予头指针初值即可。

具体实现代码如下：

class Solution {
public:
    Node* treeToDoublyList(Node* root) {
        if(root==NULL) return NULL;
        dfs(root);
        head->left=pre;
        pre->right=head;
        return head;
    }
    Node *pre=NULL,*head=NULL;
    void dfs(Node* root)
    {
        if(root==NULL) return;
        dfs(root->left);
        if(pre==NULL) head=root;
        else pre->right=root;
        root->left=pre;
        pre=root;
        dfs(root->right);
    }

};

 最后要记得由于是循环链表，所以要将头指针和尾指针再互指一下。

这题的关键应该是想到用一个前驱节点和利用二叉搜索树的性质。