二叉搜索树与双链表

二叉搜索树转换双链表 不新建节点 指针变换 两个方法 刷题看到的很nice 先存上

这儿是 挖井人.

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

方法一：非递归版解题思路：

1.核心是中序遍历的非递归算法。
2.修改当前遍历节点与前一遍历节点的指针指向。

class Solution {
public:
    TreeNode* Convert(TreeNode* pRootOfTree)
    {
        if(pRootOfTree==NULL) return NULL;
        vector<TreeNode*> stack;
        TreeNode* pre=NULL;
        TreeNode* p=pRootOfTree;
        TreeNode* root;
        bool first=true;
        while(p!=NULL||!stack.empty())
        {
            while(p!=NULL)
            {
                stack.push_back(p);
                p=p->left;
            }
            p=stack.back();
            stack.pop_back();
            if(first)
            {
                root=p;
                pre=p;
                first=false;
            }
            else
            {
                pre->right=p;
                p->left=pre;
                pre=p;
            }
            p=p->right;
        }
        return root;
    }
};

方法二：递归版解题思路：

1.将左子树构造成双链表，并返回链表头节点。
2.定位至左子树双链表最后一个节点。
3.如果左子树链表不为空的话，将当前root追加到左子树链表。
4.将右子树构造成双链表，并返回链表头节点。
5.如果右子树链表不为空的话，将该链表追加到root节点之后。
6.根据左子树链表是否为空确定返回的节点。

class Solution {
public:
    TreeNode* Convert(TreeNode* pRootOfTree)
    {
        if(pRootOfTree==NULL) return NULL;
        if(pRootOfTree->left==NULL&&pRootOfTree->right==NULL) return pRootOfTree;
        TreeNode* left=Convert(pRootOfTree->left);
        TreeNode* p=left;
        while(p!=NULL&&p->right!=NULL)
        {
            p=p->right;
        }
        if(left!=NULL)
        {
            pRootOfTree->left=p;
            p->right=pRootOfTree;
        }
        TreeNode* right=Convert(pRootOfTree->right);
        if(right !=NULL)
        {
            pRootOfTree->right=right;
            right->left=pRootOfTree;
        }
        return left!=NULL? left:pRootOfTree;
    }
};