04-树4 是否同一棵二叉搜索树(c++,二叉搜索树)

04-树4 是否同一棵二叉搜索树

分数 25

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

输入格式:

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

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

输出格式:

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

本题思路

考察就是对二叉搜索树的知识,左子树<根节点<右子树;怎么去插入,删除,这些都在《hello》算法书有;

之后,应该只要不比中序遍历,都可以过;

输入样例:

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

输出样例:

Yes
No
No
#include <bits/stdc++.h>
using namespace std;
#define int long long
#define iloves ios::sync_with_stdio(false);cin.tie(0);cout.tie(0);
int n,m,a,len1,len2;
struct TreeNode{
    int val;
    TreeNode *left,*right;
    TreeNode(int x):val(x),left(nullptr),right(nullptr){}
};
TreeNode *root;
void insert(int num){
    if(root==nullptr){root=new TreeNode(num);return;}
    TreeNode *cur=root,*pre=nullptr;
    while(cur!=nullptr){
        if(cur->val==num)return;
        pre=cur;
        if(cur->val>num)cur=cur->left;
        else cur=cur->right;
    }TreeNode* node=new TreeNode(num);
    if(pre->val>num)pre->left=node;
    else pre->right=node;
}
vector<int>levelOrder(TreeNode* node){
    vector<int>vec;
    stack<TreeNode*>st;
    st.push(node);
    while(!st.empty()){
        TreeNode *p=st.top();
        st.pop();
        vec.push_back(p->val);
        if(p->left!=nullptr)st.push(p->left);
        if(p->right!=nullptr)st.push(p->right);
    }
    return vec;
};

void slove() {
    while(1){
        cin>>n;
        if(n==0)break;
        else cin>>m;
        for(int i=1;i<=n;i++){
            cin>>a;
            insert(a);
        }
        vector<int> vc=levelOrder(root);
        root=nullptr;
        while(m--){
            for(int i=1;i<=n;i++){
                cin>>a;
                insert(a);
            }
            vector<int>vc2=levelOrder(root);
            root=nullptr;
            len1=vc.size();
            len2=vc2.size();
            a=0;
            if(len1!=len2)a=-1;
            else{
                for(int i=0;i<len1;i++){
                    if(vc[i]!=vc2[i]){a=-1;break;}
                }
            }
            if(a==-1)cout<<"No"<<endl;
            else cout<<"Yes"<<endl;
        }
    }
}
signed main() {
    iloves;
    int test = 1;
    while (test--) {
        slove();
    }
}
  • 6
    点赞
  • 4
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值