【id:129】【20分】C. DS二叉排序树之删除

【id:129】【20分】C. DS二叉排序树之删除

时间限制1s

内存限制128MB

题目描述:

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

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

输入:

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

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

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

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

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

以此类推输入下一个示例

输出:

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

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

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

输入样例1 :

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

输出样例1:

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

AC代码：

#include <iostream>

using namespace std;

class Node {
public:
    int info;
    Node *left;
    Node *right;
};

void mid_traverse(Node *p){//中序遍历
    if (p){
        mid_traverse(p->left);
        cout<<p->info<<" ";
        mid_traverse(p->right);
    }
    else return;
}

void Delete(Node *p,int key){
    Node *pre=p;//记录父节点方便后续操作
    while(p){
        if (p->info == key){
            if (!p->left && !p->right){//是叶子节点
                if (pre->left->info==key)pre->left= nullptr;
                else pre->right= nullptr;
                return;
            }
            else if(!p->left && p->right){//只有右节点
                p=p->right;
                return;
            }
            else if(p->left && !p->right){//只有左节点
                p=p->left;
                return;
            }
            else{//左右节点都有
                Node *left_tree=p->left;

                Node *rightmost_son=left_tree->right;
                if (rightmost_son){
                    while(rightmost_son->right){
                        left_tree=left_tree->right;
                        rightmost_son=rightmost_son->right;
                    }
                    p->info=rightmost_son->info;
                    Delete(left_tree,rightmost_son->info);
                    return;
                }
                else{
                    p->info=p->left->info;
                    p->left=p->left->left;
                    return;
                }
            }
        }
        else if (p->info > key){
            pre=p;
            p=p->left;
        }
        else if(p->info < key){
            pre=p;
            p=p->right;
        }
    }
}

int main() {
    int t;
    cin >> t;
    for (int i = 0; i < t; ++i) {

        int n;
        cin >> n;
        Node *root = new Node();
        for (int j = 0; j < n; ++j) {//建树过程
            if (j == 0) cin >> root->info;
            else {
                Node *node = new Node();
                cin >> node->info;

                Node *p = root;
                int flag = 0;
                while (flag == 0) {
                    if (p->info < node->info && p->right) {
                        p = p->right;
                    } else if (p->info > node->info && p->left) {
                        p = p->left;
                    } else if (p->info < node->info) {
                        p->right = node;
                        flag = 1;
                    } else if (p->info > node->info) {
                        p->left = node;
                        flag = 1;
                    }
                }
            }
        }

        mid_traverse(root);
        cout<<endl;

        int m;cin>>m;
        for (int k = 0; k < m; ++k) {
            int key;cin>>key;

            Delete(root,key);
            mid_traverse(root);
            cout<<endl;
        }
    }
}