zoukankan      html  css  js  c++  java
  • PTA-7-4 是否是同一棵二叉搜索树

    总体思路:采用先序遍历的方式来进行比较即可。(静态树)

    题目描述:

    给定一个插入序列就可以唯一确定一棵二叉搜索树。然而,一棵给定的二叉搜索树却可以由多种不同的插入序列得到。例如分别按照序列{2, 1, 3}和{2, 3, 1}插入初始为空的二叉搜索树,都得到一样的结果。于是对于输入的各种插入序列,你需要判断它们是否能生成一样的二叉搜索树。

    输入格式:

    输入包含若干组测试数据。每组数据的第1行给出两个正整数N (≤10)和L,分别是每个序列插入元素的个数和需要检查的序列个数。第2行给出N个以空格分隔的正整数,作为初始插入序列。最后L行,每行给出N个插入的元素,属于L个需要检查的序列。

    简单起见,我们保证每个插入序列都是1到N的一个排列。当读到N为0时,标志输入结束,这组数据不要处理。

    输出格式:

    对每一组需要检查的序列,如果其生成的二叉搜索树跟对应的初始序列生成的一样,输出“Yes”,否则输出“No”。

    输入样例:

    4 2
    3 1 4 2
    3 4 1 2
    3 2 4 1
    2 1
    2 1
    1 2
    0
    

    输出样例:

    Yes
    No
    No
    

    这个题目需要掌握的知识点就是二叉搜索树结点的插入与遍历

    对于一般的二叉树,我们采用如下的结构体代表结点。

    因为直接给出了节点之间的关系,在这个题中,我们直接使用静态二叉树来完成树信息的保存。

    struct Node
    {
        char data;
        int left, right, layer;
    } tree1[11], tree2[11]; // 两棵树分别保存左右两边的信息
    
    

    对于一个搜索二叉树插入结点,我们直接使用搜索二叉树的性质即可:

    Node *insert(Node *&root, int key)
    {
        if (root == NULL)
        {
            root = new Node(key);
            return root;
        }
        else if (key < root->data)
        {
            insert(root->left, key);
        }
        else
        {
            insert(root->right, key);
        }
    }
    

    当我们创建好了完整的两棵二叉树之后,我们就可以通过先序遍历的方式来进行两棵树的比较。

    这里需要注意的是,空树也是相同的!,只需要注意好这个点,我们就能很方便的比较出来,比较函数如下:

    // 先序遍历判断两棵树树否相同
    bool Judge(Node *root1, Node *root2)
    {
        if (root1 == NULL && root2 == NULL)
            return true; // 空树相同
        if ((root1 == NULL && root2 != NULL) || (root1 != NULL && root2 == NULL))
            return false; // 只要有一个存在,一个不存在就不同
        if (root1->data == root2->data)
        { // 数据相同时,比较左右子树
            bool leftSame = Judge(root1->left, root2->left);
            bool rightSame = Judge(root1->right, root2->right);
            return leftSame && rightSame;
        }else { // 数据不同直接`    1
            return false;
        }
    }
    

    当然采取别的遍历方式也可以,但是先序遍历方式是效率最高的一种。

    完整代码如下:

    /*
        Author: Veeupup
        判断是否是同一棵二叉搜索树
     */
    #include <iostream>
    #include <cstdio>
    #include <cstdint>
    #include <queue>
    using namespace std;
    
    struct Node
    {
        int data;
        Node *left, *right;
        Node(int _data) : data(_data), left(NULL), right(NULL) {}
    };
    
    int n, l; // 二叉树结点数目,需要检查的序列数
    
    Node *insert(Node *&root, int key)
    {
        if (root == NULL)
        {
            root = new Node(key);
            return root;
        }
        else if (key < root->data)
        {
            insert(root->left, key);
        }
        else
        {
            insert(root->right, key);
        }
    }
    
    // 构建树
    Node *createTree(int n)
    {
        Node *root = NULL;
        int key;
        for (int i = 0; i < n; i++)
        {
            scanf("%d", &key);
            insert(root, key);
        }
        return root;
    }
    
    // 先序遍历判断两棵树树否相同
    bool Judge(Node *root1, Node *root2)
    {
        if (root1 == NULL && root2 == NULL)
            return true; // 空树相同
        if ((root1 == NULL && root2 != NULL) || (root1 != NULL && root2 == NULL))
            return false; // 只要有一个存在,一个不存在就不同
        if (root1->data == root2->data)
        { // 数据相同时,比较左右子树
            bool leftSame = Judge(root1->left, root2->left);
            bool rightSame = Judge(root1->right, root2->right);
            return leftSame && rightSame;
        }else { // 数据不同直接`    1
            return false;
        }
    }
    
    int main()
    {
        // freopen("data.txt", "r", stdin);
        while (scanf("%d", &n) != EOF)
        {
            if (n == 0)
                break;
            scanf("%d", &l);
            Node *originTree = createTree(n); // 生成默认的树
            // levelOrder(originTree);
            for (int i = 0; i < l; i++)
            {
                Node* nowRoot = createTree(n);
                // levelOrder(nowRoot);
                if (Judge(originTree, nowRoot))
                {
                    printf("Yes
    ");
                }
                else
                {
                    printf("No
    ");
                }
            }
        }
        return 0;
    }
    
  • 相关阅读:
    c++ struct 使用
    c++数组、字符串操作
    c++ List、Vector、Stack、Queue使用
    十三、哈希表
    十二、234树
    十一、红黑树
    十、哈夫曼编码
    九、二叉树
    八、高级排序
    七、递归
  • 原文地址:https://www.cnblogs.com/veeupup/p/12592684.html
Copyright © 2011-2022 走看看