C++ JZ36 二叉搜索树转为双向链表

C++ JZ36 二叉搜索树转为双向链表,第1张

描述
输入一棵二叉搜索树,将该二叉搜索树转换成一个排序的双向class="superseo">链表。
注意:
1.要求不能创建任何新的结点,只能调整树中结点指针的指向。当转化完成以后,树中节点的左指针需要指向前驱,树中节点的右指针需要指向后继
2.返回链表中的第一个节点的指针
3.函数返回的TreeNode,有左右指针,其实可以看成一个双向链表的数据结构
4.你不用输出双向链表,程序会根据你的返回值自动打印输出
输入描述:
二叉树的根节点
返回值描述:
双向链表的其中一个头节点。
思路:获取二叉搜索树的中序遍历,构建双向链表

class Solution {
public:
    vector<int>ans;
    void mid(TreeNode* p){
        if(!p)return;
        mid(p->left);
        ans.push_back(p->val);
        mid(p->right);
    }
    void insert(TreeNode*La,TreeNode*R,TreeNode*temp){
        La->right = temp;
        temp->left = La;
        temp->right = R;
        R->left = temp;
    }
    TreeNode* Convert(TreeNode* pRootOfTree) {
        mid(pRootOfTree);
        if(ans.empty())return NULL;
        if(ans.size()==1)return pRootOfTree;
        TreeNode* L = new TreeNode(ans[0]);
        TreeNode* R = new TreeNode(ans[ans.size()-1]);        
        L->right=R;
        R->left = L;
        TreeNode* La = L;
        TreeNode* temp;
        for(int i =1;i<ans.size()-1;i++){
            temp = new TreeNode(ans[i]);
            insert(La,R,temp);
            La= temp;
        }
        return L;        
    }
};

欢迎分享,转载请注明来源:内存溢出

原文地址: https://outofmemory.cn/langs/789215.html

(0)
打赏 微信扫一扫 微信扫一扫 支付宝扫一扫 支付宝扫一扫
上一篇 2022-05-05
下一篇 2022-05-05

发表评论

登录后才能评论

评论列表(0条)

保存