树的递归操作说来说去就这么几个,这里很明显要用中序,因为要形成的是从小到大的双向链表。BST的中序就是从小到大的。
这里的中序就难在根那个地方的操作。递归函数需要一个指针的引用来作为前一个结点(pre),pre的初值为空指针。递归函数先进行左子树的递归,递归到最左的结点后,将其left改为pre(这样你就知道pre的初值为什么是空指针了,因为最左的结点的left是不可能有数字的)。如果pre不为空,就将pre的right改为当前结点。将pre改为root。我们可以想象pre是指向你自己构建的链表的最后一个元素的,它需要更新。然后递归右子树。
因为要返回链表的头结点,所以我们从根结点循环访问它的left,直到left为空,就返回当前结点
/*
struct TreeNode {
int val;
struct TreeNode *left;
struct TreeNode *right;
TreeNode(int x) :
val(x), left(NULL), right(NULL) {
}
};*/
class Solution {
public:
void inOrderTravese(TreeNode* root, TreeNode* &pre){
//传引用改变pre本身的值,返回此层递归后,上一层递归要用改变后的pre值
if(root == NULL) return;
//左:
inOrderTravese(root->left,pre);
root->left = pre;
if(pre!=NULL) pre->right = root;
pre = root;
inOrderTravese(root->right,pre);
}
TreeNode* Convert(TreeNode* pRootOfTree)
{
//BST转化为排序的双向链表,中序遍历即可
//其实树的三序遍历的递归思想主要是写出根那一步的操作
//此题需要一个前驱结点,也就是你自己在构建这条双向链表时当前的最后一个结点
//把你现在在操作的这个结点的left指向前驱结点,把前驱结点的right指向当前结点
//然后更新当前这条链表的最后一个节点
if(pRootOfTree == NULL) return NULL;
TreeNode* pre = NULL;
inOrderTravese(pRootOfTree,pre);
TreeNode* res = pRootOfTree;
while(res->left != NULL) res = res->left;
return res;
}
};