[微软面试100题] 二元查找数变成有序的双向链表

题目

输入一颗二元查找树，将该二元查找树转换成一个排序的双向链表。

要求：不能创建任何新的节点，只能调整指针的指向。

例如：

      10

   /         \

  6        14

 /  \       /     \

4  8  12   16

转换后的结果为 4--6--8--10--12--14--16

struct BSTreeNode
{
    int m_nValue;           // value of the node
    BSTreeNode *m_pLeft;    // left child
    BSTreeNode *m_pRight;   // right child
};

分析

涉及到Tree结构的题目，最方便快捷的就是递归解决。

我们知道将二元查找树按顺序输出的伪代码如下，只需稍微改动即可。

void printBStreeInOrder(BSTreeNode *node)
{
    if (node == NULL) return;
    printBStreeInOrder(node->m_pLeft);
    cout << node->m_nValue << endl;
    printBStreeInOrder(node->m_pRight);
}

代码

/**
 * @param root The root node of the tree
 * @return The head note of the list
 */
BSTreeNode *BSTreeToLinkedList(BSTreeNode *root)
{
    BSTreeNode *head, *tail;
    helper(head, tail, root);
    return head;
}

void helper(BSTreeNode *&head, BSTreeNode *&tail, BSTreeNode *root)
{
    if (root == NULL)
    {
        head = tail = NULL;
        return ;
    }
    BSTreeNode *l, *r;
    helper(head, l, root->m_pLeft);
    helper(r, tail, root->m_pRight);
    if (l != NULL)
    {
        l->m_pRight = root;
        root->m_pLeft = l;
    }
    else
    {
        head = root;
    }
    if (r != NULL)
    {
        root->m_pRight = r;
        r->m_pLeft = root;
    }
    else
    {
        tail = root;
    }
}