04-树4 是否同一棵二叉搜索树 (25分)

本文介绍了一种算法,用于判断多个插入序列是否能够生成相同的二叉搜索树。通过构建树结构并比较不同序列生成的树来实现这一目标。
04-树4 是否同一棵二叉搜索树   (25分)

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

输入格式:

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

简单起见,我们保证每个插入序列都是1到NN的一个排列。当读到NN为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

主要思路:
1、构建树的方式,我采用的方式是vector的下标为该结点的值,结点中存放该结点的左右孩子在vector中的下标
2、使用迭代的方式构建树,每输入一个新结点后,4种情况的判断(见代码)
3、比较两树是否相等,通过比较两树中值(vector下标)相同的两个的结点的左右孩子是否相同,每一对对应结点都相同则两棵树相同

#include<iostream>
#include<vector>

using namespace std;

#define Max_Node 11
#define END -1

typedef struct node
{
    int left;
    int right;
}Node;

void Initialize_Tree(vector<Node>& Tree,int N)//初始化树
{
    for (int i=1; i<=N; ++i)
    {
        Tree[i].left=END;
        Tree[i].right=END;
    }
}

void Build_Tree(vector<Node>& Tree,int N)//读取输入并建树
{
    int value,flag=0,root=0,pre=0;
    while (N--)
    {
        cin>>value;
        if (flag==0)//读到的第一个结点为根结点,记录下来
        {
            root=value;
            pre=root;
            flag=1;
        }else
        {
            while (1)
            {
                if (value>pre && Tree[pre].right!=END)//当前输入值比访问的上一个结点pre(pre最初为根结点)大,且pre有右孩子
                {
                    pre=Tree[pre].right;
                }
                if (value>pre && Tree[pre].right==END)//当前输入值比访问的上一个结点pre(pre最初为根结点)大,且pre无右孩子
                {
                    Tree[pre].right=value;
                    pre=root;//下一次输入数字也从根结点开始比较
                    break;
                }
                if (value<pre && Tree[pre].left!=END)//当前输入值比访问的上一个结点pre(pre最初为根结点)小,且pre有左孩子
                {
                    pre=Tree[pre].left;
                }
                if (value<pre && Tree[pre].left==END)//当前输入值比访问的上一个结点pre(pre最初为根结点)小,且pre无左孩子
                {
                    Tree[pre].left=value;
                    pre=root;//下一次输入数字也从根结点开始比较
                    break;
                }
            }
        }
    }
}

bool Compare_Tree(vector<Node> Tree1,vector<Node> Tree2,int N)//比较两棵树的每一个结点的左右孩子结点是否相等
{
    int i=1;
    for (; i<=N; ++i)
    {
        if (!(Tree1[i].left==Tree2[i].left && Tree1[i].right==Tree2[i].right))
        {
            return false;
        }
    }
    return true;
}

int main()
{
    int N,L;
    int flag=0;
    while (1)
    {
        cin>>N;
        if (N==0)
        {
            break;
        }
        cin>>L;
        vector<vector<Node> > vec(L,vector<Node>(Max_Node));
        vector<Node> Tree(Max_Node);
        Initialize_Tree(Tree, N);
        for (int i=0; i<L; ++i)
        {
            Initialize_Tree(vec[i], N);
        }
        Build_Tree(Tree, N);
        for (int i=0; i<L; i++)
        {
            Build_Tree(vec[i], N);
            if (Compare_Tree(Tree, vec[i], N))
            {
                if (flag==0)
                {
                    flag=1;
                    cout<<"Yes";
                }else
                {
                    cout<<'\n'<<"Yes";
                }
            }else
            {
                if (flag==0)
                {
                    flag=1;
                    cout<<"No";
                }else
                {
                    cout<<'\n'<<"No";
                }
            }
        }
    }
    
    return 0;
}

### 回答1: 题目描述: 给定两个整数序列,别表示两二叉搜索树的中序遍历序列,判断这两二叉搜索树是否同一。 解题思路: 二叉搜索树的中序遍历序列是有序的,因此我们可以将两个序列排序后比较是否相等。如果相等,则说明两二叉搜索树相同。 代码实现: ``` bool isSameBST(vector<int>& seq1, vector<int>& seq2) { if (seq1.size() != seq2.size()) { return false; } sort(seq1.begin(), seq1.end()); sort(seq2.begin(), seq2.end()); return seq1 == seq2; } ``` 时间复杂度:$O(nlogn)$ 空间复杂度:$O(n)$ 其中,$n$ 表示二叉搜索树中节点的个数。 ### 回答2: 题目描述: 给定两个长度相等的整型数组,判断它们是否对应同一二叉搜索树的后序遍历序列。 解题思路: 本题的难点在于如何判断两个数组是否对应同一二叉搜索树的后序遍历序列。 首先,我们知道二叉搜索树的后序遍历序列的最后一个元素为根节点,而且比根节点小的元素在左子中,比根节点大的元素在右子中。 因此,我们可以将数组前面的元素为左子和右子两个部,然后在子中递归检查,直到只剩下一个元素为止。 具体实现上,我们可以先找到根节点,然后将数组根据根节点为左右两个子数组,再递归检查左右子是否对应同一二叉搜索树的后序遍历序列。 如果左右子都满足条件,那么说明整个数组对应同一二叉搜索树的后序遍历序列,否则不是。 Code: ### 回答3: 题目描述: 给定两个长度为n的数组,按照给定的顺序插入到一空的二叉搜索树中,判断两个数组是否最终构建了同一二叉搜索树。 思路析: 本题是一道典型的二叉搜索树的应用题,题目要求我们判断两个经过插入操作后最终构建的是否相等。 首先回顾一下二叉搜索树的特点:对于一个二叉搜索树,它的左子中的所有节点的值都小于根节点的值,而它右子中的所有节点的值都大于根节点的值。 对于这个问题来说,我们可以在构建二叉搜索树的同时统计每个节点的左子大小来判断两是否相同。 具体来说,对于每个节点,我们可以统计它左子中的节点数目,然后每插入一个节点就更新一遍。当构建完毕后,每个节点的左子大小就都已经确定了。然后判断两个构建好的中所有节点的左子大小是否都一一对应即可。 代码实现: 为了方便起见,我们可以先实现一个二叉搜索树的插入操作,然后再利用这个代码来对两个输入排序数组进行构建。 //二叉搜索树结构体 struct TreeNode { int val; int left_size;//记录左子大小的变量 TreeNode *left; TreeNode *right; TreeNode(int x) : val(x), left_size(0), left(NULL), right(NULL) {} }; TreeNode* insert(TreeNode* root, int val){ if(!root) return new TreeNode(val); if(val < root->val){ root->left = insert(root->left, val); ++root->left_size;//更新左子大小 } else root->right = insert(root->right, val); return root; } 然后我们在主函数中对两个输入数组进行插入操作,之后再进行比较即可。 bool isSameTree(vector<int>& nums1, vector<int>& nums2) { if(nums1.size() != nums2.size()) return false; TreeNode* root1 = nullptr;//第一的根节点 TreeNode* root2 = nullptr;//第二的根节点 for(int i=0; i<nums1.size(); ++i){ root1 = insert(root1, nums1[i]); root2 = insert(root2, nums2[i]); } function<bool(TreeNode*, TreeNode*)> dfs = [&](TreeNode* node1, TreeNode* node2){ if(node1 == nullptr && node2 == nullptr) return true; else if(node1 == nullptr || node2 == nullptr) return false; else if(node1->val != node2->val || node1->left_size != node2->left_size) return false; else return dfs(node1->left, node2->left) && dfs(node1->right, node2->right); }; return dfs(root1, root2); } 时间复杂度:插入节点的时间复杂度是O(logn),所以整体时间复杂度是O(nlogn)。 空间复杂度:的深度最大为O(n),所以空间复杂度也是O(n)。 参考链接: https://leetcode-cn.com/problems/same-tree/description/ https://leetcode-cn.com/problems/two-sum-iv-input-is-a-bst/description/
评论
成就一亿技术人!
拼手气红包6.0元
还能输入1000个字符
 
红包 添加红包
表情包 插入表情
 条评论被折叠 查看
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值