1. 程式人生 > >劍指offer36--二叉搜尋樹與雙向連結串列

劍指offer36--二叉搜尋樹與雙向連結串列

題目描述

輸入一棵二叉搜尋樹,將該二叉搜尋樹轉換成一個排序的雙向連結串列。要求不能建立任何新的結點,只能調整樹中結點指標的指向。

思路:

不能建立新節點

肯定是中序遍歷,麻煩的在於需要保持一個遍歷過程中連結串列的尾巴,這樣才能遞迴成連結串列

/*
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);
    }
};