排序--二叉搜索树

// BST.cpp : 定义控制台应用程序的入口点。
//

#include "stdafx.h"
#include <stdlib.h>

/* 定义二叉搜索树的节点 */
typedef struct BiTNode{
	int data; // 值域
	struct BiTNode *lchild, *rchild; // 左右子节点
} BiTNode, *BiTree;

enum Status
{
	FALSE,
	TRUE
};

/* 
	在树中查找值为 key 的节点
	查找成功返回 TRUE，否则返回 FALSE
	parent：待查找节点的父节点
	target：用于迭代
	result：返回查找结果
	key：要查找的节点值
*/
Status SearchBST(BiTree parent, BiTree target, BiTree* result, int key)
{
	if (target == NULL)
	{
		*result = parent; // 如果没有找到，返回父节点
		return FALSE; // 没有找到，返回false
	}
	else if (key == target->data)
	{
		*result = target; // 找到了节点，返回 result
		return TRUE; 
	}
	else if (key < target->data)
	{
		return SearchBST(target, target->lchild, result, key);
	}
	else
	{
		return SearchBST(target, target->rchild, result, key);
	}
}

/* 
	插入元素 
	插入成功返回 TRUE，否则返回 FALSE
	target：插入以 target 作为根节点的树中
	key：插入的值
*/
Status InsertBST(BiTree* target, int key)
{
	BiTree insertPoint, tmp;
	if (!SearchBST(NULL, *target, &insertPoint, key))
	{
		tmp = (BiTree)malloc(sizeof(BiTNode)); // 为插入节点申请空间
		tmp->data = key; // 初始化节点值
		tmp->lchild = tmp->rchild = NULL; // 初始化节点的左右子节点
		if (insertPoint == NULL) // 如果插入点为 NULL，说明树为空，以 tmp 作为根节点
		{
			*target = tmp ;
		}
		else if (key < insertPoint->data) // 如果插入值小于插入点的值，则插入到左子节点
		{
			insertPoint->lchild = tmp;
		}
		else // 如果插入值大于插入点的值，则插入到右子节点
		{
			insertPoint->rchild = tmp;
		}
		return TRUE;
	}
	else
	{
		return FALSE;
	}
}

/* 
	在以 T 作为根节点的树中删除置为 key 的节点
	删除成功返回 TRUE，否则返回 FALSE
	key：待删除的节点的值
*/
Status DeleteBST(BiTree *T, int key)
{
	if (*T == NULL)
	{
		return FALSE; // 如果 T 为空，则返回 FALSE
	}
	else
	{
		if (key == (*T)->data)
		{
			return delete_aux(T); // 如果节点 T 的值为 key 则调用 delete_aux 删除节点 T
		}
		else if (key < (*T)->data) // 如果节点 T 的值大于 key，则待删除的节点在 T 节点的左边
		{
			return DeleteBST(&(*T)->lchild, key);
		}
		else // 如果节点 T 的值小于 key，则待删除的节点在 T 节点的右边
		{
			return DeleteBST(&(*T)->rchild, key);
		}
	}
}

Status delete_aux(BiTree* p)
{
	if ((*p)->rchild == NULL) // 如果待删除的节点没有右子节点
	{
		BiTree tmp = *p; // 缓存待删除的节点
		*p = (*p)->lchild; // 接入待删除节点的左子节点
		free(tmp); // 删除节点
	}
	else if ((*p)->lchild == NULL) // 如果待删除的节点没有左子节点
	{
		BiTree tmp = *p ; // 缓存待删除的节点
		*p = (*p)->rchild; // 接入待删除节点的右子节点
		free(tmp); // 删除节点
	}
	else
	{
		BiTree parent, child;
		parent = *p; // 缓存待删除的节点
		child = (*p)->lchild; // 找到待删除节点的左子节点
		while (child->rchild) // 找到左子节点最右边的节点
		{
			parent = child;
			child = child->rchild;
		}
		(*p)->data = child->data; // 接入左子节点最右边的节点的值
		if (parent != *p)
		{
			parent->rchild = child->lchild;
		}
		else
		{
			parent->lchild = child->lchild;
		}
		free(child);
	}
	return TRUE;
}

int _tmain(int argc, _TCHAR* argv[])
{
	int a[10] = {62, 88, 58, 47, 35, 73, 51, 99, 37, 93};
	BiTree T = NULL ;
	for (int i = 0; i < 10; ++i)
	{
		InsertBST(&T, a[i]);
	}
	return 0;
}