C/C++二叉树的删除递归法实现

时间: 2020-06-14 10:17

1. 删除函数 (递归法)

btree* Delete(btree* head, int key) {	//传入根指针和要删除的节点的值
	if (!head) {	//根指针为空, (或待删除结点不存在)
		return NULL;
	}

	if (key < head->data) {	//待删除节点的值小于目前节点的值
		head->left = Delete(head->left, key);
		return head;
	}

	if (key > head->data) {	//待删除节点的值大于目前节点的值
		head->right = Delete(head->right, key);
		return head;
	}

	if (!(head->left) && !(head->right)) {	//第一种情况, 待删除节点为叶子节点
		return NULL;
	}

	if (head->left && !(head->right)) {	//第二种情况, 待删除节点存在左节点
		return head->left;
	}

	if (!(head->left) && head->right) {	//第三种情况, 待删除节点存在左节点
		return head->right;
	}

	//第四种情况, 待删除节点存在左右子节点
	//通过  find  函数获取待删除节点的左子树上的最大值节点
	//将获取到的最大值替换待删除节点的值
	//再删除待删除节点左子树的最大值节点
  	int val = find(head->left);	
	head->data = val;
	head->left = Delete(head->left, key);
	return head;
}

2. int find(btree* head) 函数–(递归法)

int find(btree* head) {
	if (!(head->right)) {	//若找到左子树上的最大值节点, 返回它的值
		return head->data;
	}

	return find(head->right);	//若此节点非左子树的最大值节点, 则继续递归
}