2-1 是否同一棵二叉搜索树

2-1 是否同一棵二叉搜索树

分数 35
作者 陈越
单位 浙江大学

给定一个插入序列就可以唯一确定一棵二叉搜索树。然而,一棵给定的二叉搜索树却可以由多种不同的插入序列得到。例如分别按照序列{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

鸣谢青岛大学周强老师补充测试数据!
代码长度限制
16 KB
时间限制
400 ms
内存限制
64 MB
C++ (g++)

思路:

二叉查找树(Binary Search Tree),(又:二叉搜索树,二叉排序树)它或者是一棵空树,或者是具有下列性质的二叉树:
若它的左子树不空,则左子树上所有结点的值均小于它的根结点的值; 若它的右子树不空,则右子树上所有结点的值均大于它的根结点的值;
它的左、右子树也分别为二叉排序树。

建一棵树:将比对的基准序列建树,待比较序列不必建树,在树上按次序搜索待比较序列的元素,每搜到一个元素则在树上作出标记,如果在搜索序列元素的过程中,搜到了之前未搜过的元素(序列后面的元素比序列前面的元素早一步被搜到)(遍历到了未被标记的结点),则该序列与树不一致。
原理是:按照序列建成的树再按次序在树中搜索该序列中的元素时,搜索过程中不会出现序列前面的元素没搜到就搜到序列后面的元素这样的情况的。

那么我们定义一个结构体,内置变量v存放根结点的数据,左子树,右子树,flag用于标志是否被遍历到。
创建几个函数
Tree NewNoede(int v):用于建立新结点
Tree Insert(Tree T,int v):用于插入新结点。
Tree MakeTree(int N):用于生成二叉搜索树
int check(Tree T,int v):判断序列中的特定元素是否应该出现在树的对应位置
int judge(Tree T,int N):判断序列是否与树一致
void Reset(Tree T):清空树的标记
void FreeT(Tree T):释放树所占用的空间

另一种方法

代码样例:

#include<bits/stdc++.h>
using namespace std;

typedef struct TreeNode *Tree;
struct TreeNode{
    int v;
    Tree Left;
    Tree Right;
    int flag;
};

Tree NewNode(int v){
    Tree T=(Tree)malloc(sizeof(struct TreeNode));
    T->v=v;
    T->Left=T->Right=NULL;
    T->flag=0;//没有被搜索过
    return T;
}

Tree Insert(Tree T,int v){
    if(!T)//如果树为空的话,直接开辟以v为根结点的树
        T=NewNode(v);
    else{
        if(v>T->v)//如果v大于根结点,则插入右子树
            T->Right=Insert(T->Right,v);
        else
            T->Left=Insert(T->Left,v);
    }
    return T;
}

Tree MakeTree(int N){
    Tree T;
    int i,v;
    scanf("%d",&v);
    T=NewNode(v);//创建新节点
    for(i=1;i<N;i++){
        scanf("%d",&v);
        T=Insert(T,v);//做插入操作
    }
    return T;
}

int check(Tree T,int v){
    //先查看v是否与已经被访问过的结点相等
    if(T->flag){
        if(v<T->v)
            return check(T->Left,v);
        else if(v>T->v)
            return check(T->Right,v);
        else 
            return 0;//如果相等的话,返回0,错误,二叉搜索树不能出现相同结点
    }
    //再查看v是否与还未被访问过的结点相等
    else{
        if(v==T->v){//该结点之前没有被搜索过,且刚好等于这个元素
            T->flag=1;//重置标记
            return 1;//符合
        }else
            return 0;
    }
}

int Judge(Tree T,int N){//一定要将序列都遍历完,如果没有遍历完就return,则序列1剩余的元素会被当作序列2读入
    int i,v,flag=0;
    //flag=0代表目前与树还一致,1代表目前与 树不一致
    scanf("%d",&v);
    if(v!=T->v)
        flag=1;//该元素不等于树根,所以一定不一致
    else
        T->flag=1;//该元素等于树根,暂时标记为一致
    for(i=1;i<N;i++){
        scanf("%d",&v);
        //如果前面以及不一致了,则不需要检查后面的元素了,直接得到序列不一致的结论
        if(!flag&&!check(T,v))
            flag=1;
    }
    if(flag)
        return 0;
    else
        return 1;
}

void ResetT(Tree T){
    if(T->Left)
        ResetT(T->Left);
    if(T->Right)
        ResetT(T->Right);
    T->flag=0;
}

void FreeTree(Tree T){
    if(T->Left)
        FreeTree(T->Left);
    if(T->Right)
        FreeTree(T->Right);
    free(T);
}

int main(){
    int N,L,i;//N每个序列的元素个数 L序列个数
    scanf("%d",&N);
    Tree T;
    while(N){//以N为0作为数据输入截止条件
        scanf("%d",&L);
        T=MakeTree(N);
        for(i=0;i<L;i++){
         if(Judge(T,N))
                printf("Yes\n");
         else
             printf("No\n");
         ResetT(T);//清楚T中的标记flag
     }
        FreeTree(T);//要内存达到最优需要释放
        scanf("%d",&N);
    }
    return 0;
}
  • 1
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 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/
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值