【剑指offer】剑指 Offer 36. 二叉搜索树与双向链表

题目链接

剑指 Offer 36. 二叉搜索树与双向链表

思路分析

既然想变成一个排序的双向链表，其次题目中给出了一个二叉搜搜索树，那么我们遍历的方法也随之确定，要采用中序遍历，如果想要一个双向链表，那么就必须记录下前一个节点，因此还需要一个额外指针指向前一个节点，题目中还要求一个head指针指向最小元素，该节点就是链表中的第一个节点，因此判断条件就为pre == null，在每一次修改完指向关系之后，就需要更新pre的位置

代码实现

/*
// Definition for a Node.
class Node {
public:
    int val;
    Node* left;
    Node* right;

    Node() {}

    Node(int _val) {
        val = _val;
        left = NULL;
        right = NULL;
    }

    Node(int _val, Node* _left, Node* _right) {
        val = _val;
        left = _left;
        right = _right;
    }
};
*/
class Solution {
public:
    Node* pre = NULL;
    Node* head = NULL;
    void traversal(Node* root)
    {
        if(root == NULL)
            return ;
        traversal(root->left);

        if(pre == NULL)
        {
            head = root;
        }
        else
        {
            root->left = pre;
            pre->right = root;
        }
        pre = root;


        traversal(root->right);
    }
    Node* treeToDoublyList(Node* root) {
        if(root == NULL)
            return NULL;
        traversal(root);
        head->left = pre;
        pre->right = head;
        return head;
    }
};