A : DS二叉排序树之删除

A : DS二叉排序树之删除

Description

给出一个数据序列，建立二叉排序树，并实现删除功能

对二叉排序树进行中序遍历，可以得到有序的数据序列

Input
第一行输入t，表示有t个数据序列

第二行输入n，表示首个序列包含n个数据

第三行输入n个数据，都是自然数且互不相同，数据之间用空格隔开

第四行输入m，表示要删除m个数据

从第五行起，输入m行，每行一个要删除的数据，都是自然数

以此类推输入下一个示例

Output

第一行输出有序的数据序列，对二叉排序树进行中序遍历可以得到

从第二行起，输出删除第m个数据后的有序序列，输出m行

以此类推输出下一个示例的结果

Sample

Input

1
6
22 33 55 66 11 44
3
66
22
77

Output

11 22 33 44 55 66
11 22 33 44 55
11 33 44 55
11 33 44 55

Hint

当删除数据不在序列中，那么删除操作等于不执行，所以输出序列不变化

#include<iostream>
using namespace std;

struct Node {
	int data;
	Node* lchild;
	Node* rchild;
};

Node* newNode(int v) {
	Node* node = new Node;
	node->data = v;
	node->lchild = node->rchild = NULL;
	return node;
}

void insert(Node*& root, int x) {	// 用root指针的引用
	if (root == NULL) {
		root = newNode(x);
		return;
	}
	if (root->data > x) {
		insert(root->lchild, x);
	}
	else insert(root->rchild, x);
}

// 建树
Node* build(int data[], int n) {
	Node* root = NULL;
	for (int i = 0; i < n; i++) {
		insert(root, data[i]);
	}
	return root;
}

void midOrder(Node* root) {
	if (root == NULL)return;
	midOrder(root->lchild);
	cout << root->data << " ";
	midOrder(root->rchild);
}

void deleteNode(Node*& root, int k) {
	if (root == NULL) return; // 如果节点为空，什么都不做

	if (k < root->data) { // 值在左子树
		deleteNode(root->lchild, k);
	}
	else if (k > root->data) { // 值在右子树
		deleteNode(root->rchild, k);
	}
	else { // 找到了需要删除的节点
		if (root->lchild != NULL && root->rchild != NULL) { // 情况3: 两个子节点
			// 查找中序后继
			Node* succ = root->rchild;
			while (succ->lchild != NULL) {
				succ = succ->lchild;
			}
			// 替换数据
			root->data = succ->data;
			// 删除中序后继
			deleteNode(root->rchild, succ->data);
		}
		else { // 情况1或2: 一个或零个子节点
			Node* temp = root;
			if (root->lchild != NULL) {
				root = root->lchild; // 有左子节点
			}
			else {
				root = root->rchild; // 有右子节点或没有子节点
			}
			delete temp; // 删除节点
		}
	}
}

int main() {
	int t, n, del;
	cin >> t;
	while (t--) {
		cin >> n;
		int* data = new int[n];
		for (int i = 0; i < n; i++) {
			cin >> data[i];
		}
		Node* root = build(data, n);
		midOrder(root);
		cout << endl;

		cin >> n;
		while (n--) {
			cin >> del;
			deleteNode(root, del);
			midOrder(root);
			cout << endl;
		}
		delete[]data;
	}
	return 0;
}