剑指offer之二叉搜索树与双向链表

1.问题描述

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

节点的结构体：

struct TreeNode {
    int val;
    struct TreeNode *left;
    struct TreeNode *right;
    TreeNode(int x) :
            val(x), left(NULL), right(NULL) {
    }
};

2.分析

根据二叉搜素的特性，如果是中序遍历，那么遍历的结果就是从小到大的排序。
方法一：可以使用STL中的queue，一次中序遍历，把节点从小到大放到queue中。
之后从queue中提出每一个节点，改变节点的指针方向就可以了。
方法二：可以在一次中序遍历，直接改变节点的指针方向。

3.代码

//使用queue方法，获得我们需要的queue<TreeNode*>
void getOrderNode(queue<TreeNode*>& nodes,TreeNode* pRootOfTree)
{
    if(pRootOfTree)
    {
        getOrderNode(nodes,pRootOfTree->left);
        nodes.push(pRootOfTree);
        getOrderNode(nodes,pRootOfTree->right);
    }
}
//直接排序，这里之所以使用TreeNode*& ，是为了改变pre
void ConvertNode(TreeNode* pRootOfTree,TreeNode*& pre)
{
    if(pRootOfTree)
    {
        ConvertNode(pRootOfTree->left,pre);

        pRootOfTree->left = pre;
        if(pre) pre->right = pRootOfTree;
        pre = pRootOfTree;

        ConvertNode(pRootOfTree->right,pre);
    }
}
TreeNode* Convert(TreeNode* pRootOfTree)
{
    if(pRootOfTree == nullptr)
        return pRootOfTree;
    /*
    这里是方法一的使用，这里注释了
        queue<TreeNode*> nodes;
        getOrderNode(nodes,pRootOfTree);
        TreeNode* pHead = nodes.front();
        pHead->left = nullptr;
        TreeNode* pCur = pHead;
        nodes.pop();
        while(nodes.size())
        {
            TreeNode* pNode = nodes.front();
            nodes.pop();

            pNode->left = pCur;
            pCur->right = pNode;
            pCur = pCur->right;
        }
        return pHead;
        */
    TreeNode* pre = nullptr;
    ConvertNode(pRootOfTree,pre);
    TreeNode* pHead = pRootOfTree;
    //因为pRootOfTree，pre两个节点都不是头结点，需要向左移动到头结点
    while(pHead->left)
    {
        pHead = pHead->left;
    }
    return pHead;
}