HDU 3791 二叉搜索樹

二叉搜索樹

Time Limit: 2000/1000 MS (Java/Others)    Memory Limit: 32768/32768 K (Java/Others)
Total Submission(s): 2148    Accepted Submission(s): 957

Problem Description

判斷兩序列是否爲同一二叉搜索樹序列

 

 

Input

開始一個數n，(1<=n<=20) 表示有n個須要判斷，n= 0 的時候輸入結束。
接下去一行是一個序列，序列長度小於10，包含(0~9)的數字，沒有重複數字，根據這個序列能夠構造出一顆二叉搜索樹。
接下去的n行有n個序列，每一個序列格式跟第一個序列同樣，請判斷這兩個序列是否能組成同一顆二叉搜索樹。

 

 

Output

若是序列相同則輸出YES，不然輸出NO

 

 

Sample Input

2 567432

543267

576342 0

 

 

Sample Output

YES

NO

#include <stdio.h>
#include <iostream>
#include <string.h>
#include <string.h>
using namespace std;

typedef struct node
{
    char data;
    node *lchild;
    node *rchild;
    node()
    {
        lchild = rchild = NULL;
    }
}TreeNode;

void CreateTree(TreeNode *&pRoot, char ch)
{
    if (pRoot == NULL)
    {
        pRoot = new TreeNode;
        pRoot->data = ch;
    }
    else
    {
        if (ch > pRoot->data)
        {
            CreateTree(pRoot->rchild, ch);
        }
        else
        {
            CreateTree(pRoot->lchild, ch);
        }
    }    
}

bool IsEqual(TreeNode *pRoot1, TreeNode *pRoot2)
{
    if (pRoot1 != NULL && pRoot2 != NULL)
    {
        if (pRoot1->data != pRoot2->data)
        {
            return false;
        }
        if (IsEqual(pRoot1->lchild, pRoot2->lchild) && IsEqual(pRoot1->rchild, pRoot2->rchild))
        {
            return true;
        }
    }
    else
    {
        if (pRoot1 == NULL && pRoot2 == NULL)
        {
            return true;
        }
    }
    return false;
}

void DeleteTree(TreeNode *pRoot)
{
    if (pRoot != NULL)
    {
        DeleteTree(pRoot->lchild);
        DeleteTree(pRoot->rchild);
    }
    delete pRoot;
}

int main()
{
    int n;
    char str[15];
    while(scanf("%d", &n) != EOF && n != 0)
    {
        scanf("%s", str);
        TreeNode *pRoot = NULL;
        for (int i = 0; i < strlen(str); i++)
        {
            CreateTree(pRoot, str[i]);
        }
        for (int i = 0; i < n; i++)
        {
            scanf("%s", str);        
            TreeNode *pRoot1 = NULL;
            for (int j = 0; j < strlen(str); j++)
            {
                CreateTree(pRoot1, str[j]);
            }
            if (IsEqual(pRoot, pRoot1))
            {
                printf("YES\n");
            }
            else
            {
                printf("NO\n");
            }
            DeleteTree(pRoot1);
        }
        DeleteTree(pRoot);
    }
    return 0;
}