每天一道leetcode:剑指 Offer 36. 二叉搜索树与双向链表(中等&深度优先遍历&递归)

今日份题目:

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

示例

每天一道leetcode:剑指 Offer 36. 二叉搜索树与双向链表(中等&深度优先遍历&递归)_第1张图片

我们希望将这个二叉搜索树转化为双向循环链表。链表中的每个节点都有一个前驱和后继指针。对于双向循环链表,第一个节点的前驱是最后一个节点,最后一个节点的后继是第一个节点。

下图展示了上面的二叉搜索树转化成的链表。“head” 表示指向链表中有最小元素的节点。

每天一道leetcode:剑指 Offer 36. 二叉搜索树与双向链表(中等&深度优先遍历&递归)_第2张图片

题目思路

根据二叉搜索树的性质,我们知道,中序遍历可以得到树节点的从小到大的排序,那我们就在中序递归遍历的基础上改变链表。使用pre节点记录上个节点的位置,使用cur节点记录当前节点的位置,使用head节点记录链表头部的位置,然后每次让pre节点和cur节点互相指,pre左cur右,最后让head和pre节点相互指就能得到结果链表了。具体看代码注释。

补充:中序遍历

树的一种遍历方法,遍历顺序是左->根->右,如果左或右是树而非节点,那么就在子树中继续左根右,最后递归得到顺序。

int inOrder(Node* root)
{
    if(root->left) inOrder(root->left);
    return root->val;
    if(root->right) inOrder(root->right);
}

代码

/*
// 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,*head;
    void dfs(Node* cur) 
    {
        if(cur==NULL) return;
        //中序遍历
        dfs(cur->left);
        //左边pre,右边cur,二者相连,pre继续向后走直到结束
        if(pre!=NULL) pre->right=cur;
        else head=cur;
        cur->left=pre;
        pre=cur;
        dfs(cur->right);
    }

    Node* treeToDoublyList(Node* root) 
    {
        if(root==NULL) return NULL;
        dfs(root);
        //中间连好了,头尾相连
        head->left=pre;
        pre->right=head;
        return head;
    }
};

提交结果

每天一道leetcode:剑指 Offer 36. 二叉搜索树与双向链表(中等&深度优先遍历&递归)_第3张图片

欢迎大家在评论区讨论,如有不懂的部分,欢迎在评论区留言!

更新不易,宝子们点个赞支持下,谢谢!

你可能感兴趣的:(剑指Offer,深度优先,leetcode,链表,算法,职场和发展,c++,数据结构)