Leetcode刷题笔记——剑指 Offer 36. 二叉搜索树与双向链表【DFS+中序遍历】

题目

思路

显然这是一题中序遍历题，难点就是需要在中序遍历的时候处理前一个节点与当前的节点形成双向链表
我们可以用一个pre指针指向前一个节点，一开始的时候这个pre是空的，当我们找到最左边的节点时，判断pre是否为空，如果为空说明这是head节点，用head记录下来，否则说明pre指向了前一个节点，可以令pre->right = curptr，curptr->left = 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 {
private:
 Node* pre = NULL, * head=NULL;
public:
    Node* treeToDoublyList(Node* root) {
        if(root == NULL) return NULL;
        dfs(root);
        head->left = pre;
        pre->right = head;
        return head;
    }

    void dfs(Node* cur){
        if(cur==NULL) return;
        dfs(cur->left);
        if(pre == NULL){
            head = cur;
        }else{
            pre->right = cur;
        }
        cur->left = pre;	//这是难点，分两种情况，cur为最左边节点时，需要暂时令其left为空，当cur为正常节点时，cur的left应该指向前一个节点
        pre = cur;
        dfs(cur->right);
    }
};