九度OJ 1009: 二叉搜索树 两棵树的判等+二叉搜索树的建立

来源:互联网 发布:java打印所有水仙花数 编辑:程序博客网 时间:2024/05/20 16:13

    浙大的这道考研上机真题的大意是给定两个序列,判定由这两个序列所构成的二叉排序树是否一样。

    关键在于建树的过程和判定两棵树是否相等即可。判定两棵树是否相等可以用递归求解,先看当前结点是否相等或都为空,再看两棵子树的情况。

    题目URL:http://ac.jobdu.com/problem.php?id=1009

    我的AC代码,和大家分享一下。

#include<iostream>#include<stdio.h>using namespace std;struct Node{int d;Node *l, *r;Node(int data){d = data;l = r = NULL;}};bool search(Node * root, Node *& fa, int key){if(!root) return false;else{while(root){fa = root;if(key > root->d) root = root->r;else if(key < root->d) root = root->l;else return true;}return false;}}void insert(Node * &root, int key){if(!root){root = new Node(key);return;}else {Node *ins;if(!search(root, ins, key)){if(key > ins->d)ins->r = new Node(key);else ins->l = new Node(key);}}}bool treeCom(Node *s, Node *t){if(!s && !t) return true;else if(!s || !t) return false;if(s->d != t->d) return false;if(treeCom(s->l, t->l) && treeCom(s->r, t->r)) return true;}int main(){int n;Node *r, *s;char str[11];while(scanf("%d", &n) && n){r = NULL;scanf("%s", str);for(int i=0; i<10; i++)insert(r, str[i] - '0');for(int i=0; i<n; i++){s = NULL;scanf("%s", str);for(int i=0; i<10; i++)insert(s, str[i] - '0');if(treeCom(r, s)) printf("YES\n");else printf("NO\n");}} system("pause");return 0;}


原创粉丝点击