二叉排序樹 oj 2482

K_ape發表於2020-11-15

二叉排序樹

Description

二叉排序樹的定義是:或者是一棵空樹,或者是具有下列性質的二叉樹: 若它的左子樹不空,則左子樹上所有結點的值均小於它的根結點的值; 若它的右子樹不空,則右子樹上所有結點的值均大於它的根結點的值; 它的左、右子樹也分別為二叉排序樹。 今天我們要判斷兩序列是否為同一二叉排序樹

Input

開始一個數n,(1<=n<=20) 表示有n個需要判斷,n= 0 的時候輸入結束。
接下去一行是一個序列,序列長度小於10,包含(0~9)的數字,沒有重複數字,根據這個序列可以構造出一顆二叉排序樹。
接下去的n行有n個序列,每個序列格式跟第一個序列一樣,請判斷這兩個序列是否能組成同一顆二叉排序樹。(資料保證不會有空樹)
Output
YES/NO

Sample

Input
2
123456789
987654321
432156789
0

Output
NO
NO

#include <iostream>
#include <bits/stdc++.h>

using namespace std;

typedef struct node
{
    char data;
    node *l,*r;
}Tree;
bool flag;
char a[20],b[20];

Tree* Creat(Tree *root ,char x)
{
    if(!root)
    {
        root = new Tree;
        root->l = NULL;
        root->r = NULL;
        root->data = x;
    }
    else
    {
        if(x<root->data)
            root->l = Creat(root->l,x);
        else
            root->r = Creat(root->r,x);
    }
    return root;
}

void Judge(Tree *root1,Tree *root2)
{
    if((!root1 && root2) || (root1 && !root2))
    {
        flag = false;
        return ;
    }
    if(root1 && root2)
    {
        if(root1->data != root2->data)
        {
            flag = false;
            return ;
        }
        Judge(root1->l,root2->l);
        Judge(root1->r,root2->r);
    }
}

int main()
{
    ios::sync_with_stdio(false);

    int n,len;
    while(cin>>n && n!=0)
    {
        Tree *root1,*root2;
        root1 = NULL;
        cin>>a;
        len = strlen(a);
        for(int i=0; i<len; i++)
            root1 = Creat(root1,a[i]);

        while(n--)
        {
            root2 = NULL;
            flag = true;
            cin>>b;
            for(int i=0; i<len; i++)
                root2 = Creat(root2,b[i]);
            Judge(root1,root2);
            if(flag)
                cout<<"YES\n";
            else
                cout<<"NO\n";
        }
    }
    return 0;
}

相關文章