算法题24 二叉树的中序遍历之二叉搜索树转排序双向链表

题目

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

分析

　　二叉搜索树的左孩子比父节点的值小，右孩子比父节点的值大，通过中序遍历即可得到一个排序的队列。在遍历的过程中调整节点指针的指向就可以将它转换成一个排序的双向链表了

　　此前曾分析过二叉树的中序遍历的非递归方法。借此直接转化为求排序双向链表的方法

代码

TreeNode* BinaryTreeConvert2(TreeNode* root)
{
    if (!root)
    {
        throw std::exception("this is an empty tree!");
    }

    stack<TreeNode*> s_tree;
    TreeNode* node=root,*pre_node=NULL,*head=NULL;
    while (node||!s_tree.empty())
    {
        while (node)
        {
            s_tree.push(node);
            node=node->pLeft;
        }

        node=s_tree.top();
        s_tree.pop();

        /*
         *将中序遍历的输出变为指针调整
        */
        //cout<<node->value<<endl;
        //指定双向链表的头节点
        if(!head)
            head=node;

        //前后指针
        node->pLeft=pre_node;
        if (pre_node)
        {
            pre_node->pRight=node;
        }
        //记录前一个节点
        pre_node=node;


        node=node->pRight;
    }

    return head;
}

 

创建一个新的双向链表

DListNode* BinaryTreeConvert(TreeNode* root)
{
    if (!root)
    {
        throw std::exception("this is an empty tree!");
    }

    stack<TreeNode*> s_tree;
    stack<DListNode*> s_list;
    TreeNode* node=root;
    DListNode *head=NULL,*tail=NULL;
    DListNode *list_node=NULL,*pre_node=NULL;
    while (node||!s_tree.empty())
    {
        while (node)
        {
            s_tree.push(node);
            node=node->pLeft;
        }

        node=s_tree.top();
        s_tree.pop();

        /*
         *将中序遍历的输出变为双向链表节点创建以及指针调整
        */
        //cout<<node->value<<endl;
        list_node=new DListNode();
        list_node->value=node->value;
        list_node->pNext=NULL;
        list_node->pPre=NULL;

        //指定双向链表的头节点
        if(!head)
            head=list_node;

        //前后指针
        list_node->pPre=pre_node;
        if (pre_node)
        {
            pre_node->pNext=list_node;
        }

        pre_node=list_node;


        node=node->pRight;
    }

    return head;
}