数据结构——二叉排序树的删除操作（非递归）

1、被删除的节点是叶子节点
将双亲节点中相应的指针域的值改为空

2、被删除的节点只有左子树或右子树
将要删除的节点的双亲节点相应指针域的值指向被删除节点的左子树或者右子树

3、被删除节点既有左子树又有右子树
将左子树中的最大值或者右子树中的最小值代替该节点

TreeNode * DeleteTreeNode(TreeNode *root,int data){
	if(!root){
		return NULL;
	}
	TreeNode *node = NULL;//遍历指针
	TreeNode *parent = NULL;//记录指针的父节点
	node = root;
	while(1)//寻找要删除的位置
	{
        parent = node;//记录指针的父节点
		if(node ->data > data)//左边找
		{
            node = node ->left;
		}
		else if(node ->data < data)//右边找
		{
            node = node ->right;
		}
		if(node -> data == data)//找到了，退出循环
		{
			break;
		}
	}
	if(node == NULL){
		return NULL;
	}
	if(node -> left && node -> right)//左右孩子都有
	{
		TreeNode *temp = node;
        parent = node;
        node = node -> right;
		while(node -> left)//找右子树上最小的
		{
            parent = node;
            node = node -> left;
		}
        temp -> data = node -> data;//交换数据
		if(parent -> left == node)
		{
            parent ->left = node -> right;
			free(node);
		}
		else if(parent -> right == node)
		{
            parent -> right = node -> right;
			free(node);
		}
		return root;
	}
	if(!node -> left && !node -> right)//左右孩子都没有
	{
		if(node == root)//删除节点为根
		{
			free(root);
			return NULL;
		}
		if(parent -> left == node)
		{
            parent -> left = NULL;
			free(node);
		}
		else if(parent -> right == node)
		{
            parent -> right = NULL;
			free(node);
		}
		return root;
	}
	if(node -> left && !node -> right)//只有左孩子，没有右孩子
	{
		if(root == node)
		{
			root = node -> left;
            node -> left = NULL;
			free(node);
			return root;
		}
		if(parent -> left == node)
		{
            parent -> left = node -> left;
            node -> left = NULL;
			free(node);
		}
		else if(parent -> right == node)
		{
            parent -> right = node -> left;
            node -> left = NULL;
			free(node);
		}
		return root;
	}
	if(!node -> left && node -> right)//只有右孩子，没有左孩子
	{
		if(root == node)
		{
			root = node -> right;
            node -> right = NULL;
			free(node);
			return root;
		}
		if(parent -> left == node)
		{
            parent -> left = node -> right;
            node -> right = NULL;
			free(node);
		}
		else if(parent -> right == node)
		{
            parent -> right = node -> right;
            node -> right = NULL;
			free(node);
		}
		return root;
	}
}

该方法存在的问题：
①当要删除的元素不存在时，while会陷入死循环
②二叉排序树只剩下最后一个元素时无法删除

完整代码

#include <stdio.h>
#include <stdlib.h>
#define TRUE 1
#define FALSE 0
typedef struct TreeNode{
    int data;
    struct TreeNode *left,*right;
}TreeNode;
static TreeNode* preNode = NULL;
TreeNode* InsertNode(TreeNode* root,int data){//排序二叉树中插入元素
    if(!root){
        root = (TreeNode*)malloc(sizeof(TreeNode));
        root->data = data;
        root->right = NULL;
        root->left = NULL;
        return root;
    }else{
        if(root->data > data){
            root->left = InsertNode(root->left,data);
        }else if(root->data < data){
            root->right = InsertNode(root->right,data);
        }else if(root->data = data){
            return root;
        }
    }
    return root;
}
void InOrderTree(TreeNode* root){//中序遍历二叉树
    if(root == NULL){
        return;
    }
    InOrderTree(root->left);
    printf("%d->",root->data);
    InOrderTree(root->right);
}
//删除节点
TreeNode * DeleteTreeNode(TreeNode *root,int data){
	if(!root){
		return NULL;
	}
	TreeNode *node = NULL;//遍历指针
	TreeNode *parent = NULL;//记录指针的父节点
	node = root;
	while(1)//寻找要删除的位置
	{
        parent = node;//记录指针的父节点
		if(node ->data > data)//左边找
		{
            node = node ->left;
		}
		else if(node ->data < data)//右边找
		{
            node = node ->right;
		}
		if(node -> data == data)//找到了，退出循环
		{
			break;
		}
	}
	if(node -> left && node -> right)//左右孩子都有
	{
		TreeNode *temp = node;
        parent = node;
        node = node -> right;
		while(node -> left)//找右边最小的
		{
            parent = node;
            node = node -> left;
		}
        temp -> data = node -> data;//交换数据
		if(parent -> left == node)
		{
            parent ->left = node -> right;
			free(node);
		}
		else if(parent -> right == node)
		{
            parent -> right = node -> right;
			free(node);
		}
		return root;
	}
	if(!node -> left && !node -> right)//左右孩子都没有
	{
		if(node == root)//删除节点为根
		{
			free(root);
			return NULL;
		}
		if(parent -> left == node)
		{
            parent -> left = NULL;
			free(node);
		}
		else if(parent -> right == node)
		{
            parent -> right = NULL;
			free(node);
		}
		return root;
	}
	if(node -> left && !node -> right)//只有左孩子，没有右孩子
	{
		if(root == node)
		{
			root = node -> left;
            node -> left = NULL;
			free(node);
			return root;
		}
		if(parent -> left == node)
		{
            parent -> left = node -> left;
            node -> left = NULL;
			free(node);
		}
		else if(parent -> right == node)
		{
            parent -> right = node -> left;
            node -> left = NULL;
			free(node);
		}
		return root;
	}
	if(!node -> left && node -> right)//只有右孩子，没有左孩子
	{
		if(root == node)
		{
			root = node -> right;
            node -> right = NULL;
			free(node);
			return root;
		}
		if(parent -> left == node)
		{
            parent -> left = node -> right;
            node -> right = NULL;
			free(node);
		}
		else if(parent -> right == node)
		{
            parent -> right = node -> right;
            node -> right = NULL;
			free(node);
		}
		return root;
	}
}

int main() {
    int data[11] = {38,12,56,6,34,40,98,3,13,78,17};
    TreeNode* root = NULL;
    for(int i = 0; i < (sizeof(data)/sizeof (data[0]));i++){
        printf("第%d次插入的元素为：%d\n",i+1,data[i]);
        root =InsertNode(root,data[i]);
    }
    printf("\n");
    InOrderTree(root);
    printf("END\n");
    DeleteTreeNode(root,12);
    InOrderTree(root);
    printf("END\n");
    DeleteTreeNode(root,34);
    InOrderTree(root);
    printf("END\n");
    DeleteTreeNode(root,40);
    InOrderTree(root);
    printf("END\n");
    return 0;
}

程序运行结果

第1次插入的元素为：38
第2次插入的元素为：12
第3次插入的元素为：56
第4次插入的元素为：6
第5次插入的元素为：34
第6次插入的元素为：40
第7次插入的元素为：98
第8次插入的元素为：3
第9次插入的元素为：13
第10次插入的元素为：78
第11次插入的元素为：17

3->6->12->13->17->34->38->40->56->78->98->END
3->6->13->17->34->38->40->56->78->98->END
3->6->13->17->38->40->56->78->98->END
3->6->13->17->38->56->78->98->END

对应的二叉树为：