要把一棵二叉树转变成双向链表,二叉树和双向链表很类似,每个节点都是包括一个值和两个指针。针对这题,思路还是递归,选择的是树的中序遍历,中序遍历是按照从小到大的顺序遍历二叉搜索树的每一个节点,得出来的序列是有序的,注意只有二叉搜索树才有这个特性。
这道题最重要的过程在于调整原来树节点的指针,使得当前节点的左指针指向前一个节点,右指针指向后一个节点。重要的是要记录遍历到的当前节点在跳帧后链表中的位置。
代码:
TreeNode* Convert(TreeNode* pRootOfTree)
{
TreeNode* last = NULL;
ConvertCore(pRootOfTree,&last);
TreeNode* head = last;
while(head && head->left){
head = head->left;
}
return head;
}
void ConvertCore(TreeNode* pRoot,TreeNode **last){
if(!pRoot) return;
TreeNode* pCur = pRoot;//记录当前节点
if(pCur->left)//遍历左子树
ConvertCore(pCur->left,last);
pCur->left = *last;
if(*last)
(*last)->right = pCur;
*last = pCur;
if(pCur->right)
ConvertCore(pCur->right,last);
}