【牛客网】KY124 二叉搜索树

【牛客网】KY124 二叉搜索树,第1张

【牛客网】KY124 二叉搜索树


判断是否为同一二叉搜索树,首先就是要构造成二叉搜索树,则若是数据是一样的,那么中序遍历一定是一样的,因此中序遍历无法判断是否为同一二叉搜索树,但是其他遍历方式如先序遍历、后序遍历和层序遍历,则都能够判断,因此我采用的是先序遍历进行判断:

#include
#include

using namespace std;

struct TreeNode{
    int data;
    struct TreeNode *lchild, *rchild;
    TreeNode(int d): data(d), lchild(nullptr), rchild(nullptr){}
};

TreeNode* insert(TreeNode *root, int x){
    if(root == nullptr) root = new TreeNode(x);
    else if(x < root->data) root->lchild = insert(root->lchild, x);
    else if(x > root->data) root->rchild = insert(root->rchild, x);
    return root;
}

bool judge(TreeNode *root, TreeNode *root1){
    if(root == nullptr && root1 == nullptr) return true;
    if(root->data != root1->data) return false;
    else return judge(root->lchild, root1->lchild) && judge(root->rchild, root1->rchild);
}

int main(){
    int n;
    string str, str1;
    while(scanf("%d", &n) != EOF){
        if(n == 0) break;
        cin>>str;
        int k = str.size();
        TreeNode *root = nullptr;
        for(int i = 0;i < k;i++) root = insert(root, str[i] - '0');
        while(n--){
            cin>>str1;
            k = str1.size();
            TreeNode *root1 = nullptr;
            for(int i = 0;i < k;i++) root1 = insert(root1, str1[i] - '0');
            if(judge(root, root1)) printf("YESn");
            else printf("NOn");
        }
    }
    return 0;
}

欢迎分享,转载请注明来源:内存溢出

原文地址: http://outofmemory.cn/zaji/5718461.html

(0)
打赏 微信扫一扫 微信扫一扫 支付宝扫一扫 支付宝扫一扫
上一篇 2022-12-17
下一篇 2022-12-17

发表评论

登录后才能评论

评论列表(0条)

保存