HDU 3179 二叉搜索树（树的建立）

二叉搜索树

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

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

 

 

Source

浙大计算机研究生复试上机考试-2010年

 

 

Recommend

notonlysuccess   |   We have carefully selected several similar problems for you:  3787 3790 3789 3788 1710 

 

我的思路：

建一棵树，上面放有flag，当有其他数组近来匹配时，若经过的节点的flag为0则，不相等，否则，将与该数相等的树的节点的flag设为1。

 

 

可以这样建树

struct tree                //声明树的结构
{
    struct tree *left;
    int data;
    struct tree *right;
};

 

本题代码

#include <iostream>
#include <cstring>
#include <string>
#include <algorithm>
typedef struct treenode *tree;//定义treenode这个结构体是一个指针
struct treenode
{
    int v;
    tree left, right;//这个结构体指针又含有两个指针
    bool f;//用于标记该节点曾经有没有经过
};
using namespace std;

tree newnode(int x)//新建节点
{
    tree t = (tree)malloc(sizeof(treenode));
    t->v = x;
    t->left = t->right = NULL;
    t->f = 0;
    return t;//要有返回值
}

tree insert(tree t,int x)//插入节点
{
    if (!t) t = newnode(x);//如果节点为空，建一个节点
    else
    {
        if (x < t->v)
        {
            t->left = insert(t->left, x);//不能写成t=。。。因为是插入在左边
        }
        else
            t->right = insert(t->right, x);
    }
    return t;//要有返回值
}

bool check(tree t, int x)
{
    if (!t) return 0;//x在树t中没有出现过
    else
    {
        if (t->v == x)//相等，标记经过该节点
        {
            t->f = 1;
            return 1;
        }
        else
        {
            if (t->f == 0) return 0;//经过了一个曾经没经过的节点
            else
            {
                if (x < t->v)
                    return check(t->left, x);
                else
                    return check(t->right, x);
            }
        }
    }
}

void init(tree t)//初始化树上的f
{
    if (t)
    {
        t->f = 0;
        init(t->left);
        init(t->right);
    }
}

int main()
{
    int n;
    char a[20];
    while (cin >> n && n)
    {
        tree t = (tree)malloc(sizeof(treenode));//开辟空间建第一个节点
        cin >> a;
        int l = strlen(a);
        t->v = a[0]-'0';
        t->left = t->right = NULL;
        int i;
        for (i = 1; i < l; i++)
        {
            int x = a[i] - '0';
            t = insert(t,x);//插入节点；
        }
        while (n--)
        {
            cin >> a;
            bool f = 1;
            for (i = 0; i < l; i++)
            {
                int x = a[i] - '0';
                f = check(t, x);//进入树t检查
                if (!f) break;
            }
            if (!f) cout << "NO" << endl;
            else cout << "YES" << endl;
            init(t);//把树上的f重新设为0
        }
        free(t);//释放树
    }
    return 0;
}