PTA 二叉搜索树的删除操作 (C 语言)

给出一棵二叉搜索树（没有相同元素）, 请输出其删除部分元素之后的层序遍历序列。

删除结点的策略如下：

• 如果一个结点是叶子结点，则直接删除；
• 如果一个结点的左子树不为空, 则将该结点的值设置为其左子树上各结点中的最大值，并继续删除其左子树上拥有最大值的结点；
• 如果一个结点的左子树为空但右子树不为空，则将该结点的值设置为其右子树上各结点中的最小值，并继续删除其右子树上拥有最小值的结点。

输入格式:

每个输入文件包含一个测试用例。每个测试用例的第一行包含一个整数 N (0<N<=100)，表示二叉搜索树中结点的个数。
第二行给出该二叉搜索树的先序遍历序列，由 N 个整数构成，以一个空格分隔。第三行给出一个整数K (0<K<N)，表示待删除的结点个数。最后一行给出 K 个整数，表示待删除的各个结点上的值。必须按输入次序删除结点。题目保证结点一定能被删除。

输出格式:

在一行中输出删除结点后的层序遍历序列。序列中的数字以一个空格分隔，行末不得有多余空格。

输入样例:

7
4 2 1 3 6 5 7
2
3 6

 输出样例:

4 2 5 1 7 

思路 ：

        其实也没什么思路，根据排序二叉树的删除节点方法来删除

        题目虽说前序的方式输入数据建立排序二叉树，但还是和平时插入建立二叉树没什么区别

        以样例为演示排序二叉树的删除节点

        

         先删除3节点

        为了保证排序二叉树中序遍历的递增，从而需要在该节点右子树中找到最小值来代替(这只是  平时情况现根据题目要求来判断)

        此时要做的事将3的右子树中找到最大值但此时3为叶子节点直接删除

         将6删除

        此时6左右节点都有，由题意知应要在左子树里找到最大值，现在只有5满足

         对删除节点重新赋值，删除左子树中5节点

        此时5为叶子节点直接删除

#include<stdio.h>
#include<stdlib.h>

//建树的结构体
typedef struct Tree {
    int data;
    struct Tree* left;
    struct Tree* right;
}tree;

//建树
tree* buildTree(tree* T, int num) {
    if (!T) {//为空直接开辟空间
        T = (tree*)malloc(sizeof(tree));
        T->data = num;
        T->left = NULL;
        T->right = NULL;
    } else if (num < T->data) {//判断num应该在左子树还是右子树
        T->left = buildTree(T->left, num);
    } else if (num > T->data) {
        T->right = buildTree(T->right, num);
    }
    return T;
}

//用伪循环队列来进行层序遍历
void HierarchicalTraversal(tree* T) {
    tree* queue[100];//队列

    int front;//队列头
    int tail;//队列尾
    front = tail = 0;
    
    int flag = 1;//判断是否输出空格

    queue[tail++] = T;//直接将开始节点加入队列中

    while (front != tail) {

        tree* cur = queue[front++];//出队头

        //打印
        if (flag) {
            printf("%d", cur->data);
            flag = 0;
        } else {
            printf(" %d", cur->data);
        }
        
        //判断左右子树是否存在后 加入节点
        if (cur->left) {
            queue[tail++] = cur->left;
        }
        if (cur->right) {
            queue[tail++] = cur->right;
        }
    }
}

//找树中的最小值
tree* findMin(tree* T) {
    if (!T) {
        return NULL;
    }
    if (!T->left) {
        return T;
    } else {
        findMin(T->left);
    }
}

//找树中的最大值
tree* findMax(tree* T) {
    if (!T) {
        return NULL;
    }
    if (!T->right) {
        return T;
    } else {
        findMax(T->right);
    }
}

//删除节点
tree* DeleteNode(tree* T, int num) {
    tree* temp = NULL;//建临时节点到时free掉
    
    //判空
    if (!T) {
        return NULL;
    } else {
        //找num存在的节点
        if (num < T->data) {
            T->left = DeleteNode(T->left, num);
        } else if (num > T->data) {
            T->right = DeleteNode(T->right, num);
        } else {
            //根据题意得删除的几个条件
            /*
                如果一个结点是叶子结点，则直接删除；

                如果一个结点的左子树不为空, 则将该结点的值设置为其左子树上各结点中的最大值，并                                
                继续删除其左子树上拥有最大值的结点；
                
                如果一个结点的左子树为空但右子树不为空，则将该结点的值设置为其右子树上各结点中 
                的最小值，并继续删除其右子树上拥有最小值的结点。
            */
            //此时可知只要左子树存在就将左子树的最大值放进当前节点
            //向下找最大值节点
            if ((T->left && T->right) || T->left) {
                
                temp = findMax(T->left);
                T->data = temp->data;
                T->left = DeleteNode(T->left, T->data);
                
            } 
            //此时可知左子树为空右子树存在就将右子树的最小值放进当前节点
            //向下找最小值节点
            else if (T->right) {

                temp = findMin(T->right);
                T->data = temp->data;
                T->right = DeleteNode(T->right, T->data);
                
            } 
            //删除free掉
            else {
                temp = T;
                T = NULL;
                free(temp);
            }
        }
    }
    return T;
}

int main () {
    int N;
    scanf("%d", &N);
    tree* T = NULL;
    while (N--) {
        int num;
        scanf("%d", &num);
        T = buildTree(T, num);
    }
    
    int K;
    scanf("%d", &K);
    while (K--) {
        int num;
        scanf("%d", &num);
        T = DeleteNode(T, num);
    }
    
    HierarchicalTraversal(T);
    
    return 0;
}

如果还是不熟悉可以找搜索二叉树的节点删除细看