二叉搜索树与双向链表

题目描述

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

思路一：（较麻烦）

1、当前结点有左子树，则当前结点连接到左子树递归产生的双项链表的最后，返回本节点；

2、当前结点有右子树，则当前结点连接到右子树递归产生的双项链表的最前，返回本节点；

3、叶子结点直接返回本节点；

注意：由于都是返回的本节点，所以需要寻找子链表最前最后的结点，最后返回时也需要寻找；

代码：

    TreeNode* ConvertNode(TreeNode* pRootOfTree){
        if(pRootOfTree == NULL)
            return NULL;
        TreeNode* left_Tree;
        TreeNode* right_Tree;
        if(pRootOfTree->left){
            left_Tree = ConvertNode(pRootOfTree->left);
            while(left_Tree->right){
                left_Tree = left_Tree->right;
            }
            left_Tree->right = pRootOfTree;
            pRootOfTree->left = left_Tree;
        }
        if(pRootOfTree->right){
            right_Tree = ConvertNode(pRootOfTree->right);
            while(right_Tree->left){
                right_Tree = right_Tree->left;
            }
            right_Tree->left = pRootOfTree;
            pRootOfTree->right = right_Tree;
        }
        return  pRootOfTree;
    }
public:
    TreeNode* Convert(TreeNode* pRootOfTree)
    {
         if(pRootOfTree == NULL)
             return NULL;
         TreeNode* node = ConvertNode(pRootOfTree);
        while(node->left){
            node = node->left;
        }
        return node;
    }

思路二：

先中序遍历存储各个节点，然后在修改各个节点的指针关系；

代码：

TreeNode* Convert(TreeNode* pRootOfTree)
    {
        // 基本条件判断
        if(pRootOfTree == NULL) {
            return NULL;
        }
        // 中序遍历，存储节点
        vector<TreeNode*> inOrder;
        traverseInorder(pRootOfTree, inOrder);
        // 修改指针关系
        for(int i=0; i<inOrder.size()-1; i++) {
            inOrder[i]->right = inOrder[i+1];
        }
        for(int i=inOrder.size()-1; i>0; i--) {
            inOrder[i]->left = inOrder[i-1];
        }
        return inOrder[0];
    }
 
void traverseInorder(TreeNode* pRootOfTree, vector<TreeNode*>& inOrder)
    {
        if(pRootOfTree == NULL) {
            return;
        }
        traverseInorder(pRootOfTree->left, inOrder);
        inOrder.push_back(pRootOfTree);
        traverseInorder(pRootOfTree->right, inOrder);
    }