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

中序遍历+递归,建立两个节点:pre和head。

判断是否是第一个头节点,也就是pre=nullptr,用head记录第一个节点。

然后常规操作:pre->right=cur; cur->left=pre;更新pre=cur;

最后head->left=pre,pre->right=head;此时的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:
    void dfs(Node*cur)
    {
        if(cur==nullptr) return;
        
        dfs(cur->left);

        if(pre==nullptr){
            head=cur;
            cur->left=nullptr;
        }
        else{
            cur->left=pre;
            pre->right=cur;
        }
        pre=cur;

        dfs(cur->right);

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

        return head;
    
    }
private:
      Node*head,*pre=nullptr;
};

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值