剑指offer36--二叉搜索树与双向链表

版权声明:本文为博主原创文章,未经博主允许不得转载。 https://blog.csdn.net/m0_37561165/article/details/82559557

题目描述

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

思路:

不能创建新节点

肯定是中序遍历,麻烦的在于需要保持一个遍历过程中链表的尾巴,这样才能递归成链表

/*
struct TreeNode {
	int val;
	struct TreeNode *left;
	struct TreeNode *right;
	TreeNode(int x) :
			val(x), left(NULL), right(NULL) {
	}
};*/
class Solution {
public:
    TreeNode* Convert(TreeNode* pRootOfTree)
    {
        TreeNode* pLastNode=NULL;
        ConvertCore(pRootOfTree,&pLastNode);
        TreeNode* pHead=pLastNode;
        while(pHead!=NULL&&pHead->left!=NULL)
            pHead=pHead->left;
        return pHead;
    }
    void ConvertCore(TreeNode* pNode,TreeNode** pLastNode)
    {
        if(pNode==NULL)
            return;
        TreeNode* pTmp=pNode;
        if(pTmp->left!=NULL)
            ConvertCore(pTmp->left,pLastNode);
        pTmp->left=*pLastNode;//之前的结尾与本次的node互相链接
        if(*pLastNode!=NULL)
            (*pLastNode)->right=pTmp;
        *pLastNode=pTmp;
        if(pTmp->right!=NULL)
            ConvertCore(pTmp->right,pLastNode);
    }
};

猜你喜欢

转载自blog.csdn.net/m0_37561165/article/details/82559557