二進制查找樹轉換爲雙向鏈表

原創

2020-02-21 21:25

完全按照海濤哥劍指offer裏邊的遞歸思路來寫的，基本一樣，僅作學習驗證，努力鍛鍊，努力學習！

題目：輸入一棵二元查找樹，將該二元查找樹轉換成一個排序的雙向鏈表。要求不能創建任何新的結點，只調整指針的指向。

　　比如將二元查找樹
                                            10
                                          /    \
                                        6       14
                                      / \     /　 \
                                   　4     8 12 　 16
轉換成雙向鏈表

4=6=8=10=12=14=16。

code如下：

//Change a BSTree to a sorted double linklist
struct BSTreeNode 
{
	int value;
	BSTreeNode *left;
	BSTreeNode *right;
}head;
//Create a node of BSTree for test
BSTreeNode* CreateNode(int value)
{
	BSTreeNode *node = new BSTreeNode();
	node->value = value;
	node->left = NULL;
	node->right = NULL;
	return node;
}
//using a lastNode pointer to convert the pointer of the node
void ConvertNode(BSTreeNode *head, BSTreeNode **lastNode)
{
	if(head == NULL)
		return;
	BSTreeNode *pCurrent = head;
	if(pCurrent->left != NULL)
		ConvertNode(pCurrent->left, lastNode);
	pCurrent->left = *lastNode;
	if(*lastNode != NULL)
		(*lastNode)->right = pCurrent;
	*lastNode = pCurrent;
	if(pCurrent->right != NULL)
		ConvertNode(pCurrent->right, lastNode);
}
//lastNode pointer is pointing to the rear,so the head can be obtained 
BSTreeNode* Convert(BSTreeNode *head)
{
	if(head == NULL)
		return NULL;
	BSTreeNode *lastNode = NULL;
	ConvertNode(head, &lastNode);
	while(lastNode->left != NULL)
		lastNode = lastNode->left;
	return lastNode;
}
//main function for test
int main()
{
	BSTreeNode *head = CreateNode(10);
	head->left = CreateNode(8);
	head->left->left = CreateNode(7);
	head->left->right = CreateNode(9);
	head->right = CreateNode(12);
	head->right->left = CreateNode(11);
	head = Convert(head);
	while(head != NULL)
	{
		printf("%d ",head->value);
		head = head->right;
	}
}