7-4-是否同一棵二叉搜索树-编程题

解题代码

#include<stdio.h>
#include<stdlib.h>
typedef struct TNode* AVLTree;
struct TNode {
	int Data;
	AVLTree Left, Right;
	int Flag;
};
typedef enum{false,true} bool;
AVLTree NewAVLTree(AVLTree T, int N);
AVLTree Insert(AVLTree T, int K);
bool Judge(AVLTree T, int N);
bool check(AVLTree T, int K);
void Clear(AVLTree T);
int main()
{
	int N;//number of TNode
	scanf("%d", &N);
	int flag = 1;
	while (N) {
		int L;//number of line
		scanf("%d", &L);
		AVLTree T = NULL;
		T = NewAVLTree(T, N);
		while (L--) {
			if (Judge(T,N)) {
				if (flag) flag = 0;
				else printf("\n");
				printf("Yes");
			}
			else {
				if (flag) flag = 0;
				else printf("\n");
				printf("No");
			}
			Clear(T);
		}
		scanf("%d", &N);
	}
	return 0;
}
AVLTree NewAVLTree(AVLTree T,int N) {
	int i, temp;
	for (i = 0; i < N; i++) {
		scanf("%d", &temp);
		T = Insert(T, temp);
	}
	return T;
}
AVLTree Insert(AVLTree T, int K) {
	if (!T) {
		T = malloc(sizeof(struct TNode));
		T->Data = K;
		T->Left = T->Right = NULL;
		T->Flag = 0;
	}
	else if (K < T->Data) T->Left = Insert(T->Left, K);
	else if (K > T->Data) T->Right = Insert(T->Right, K);
	return T;
}
bool Judge(AVLTree T,int N) {
	int temp, i, flag=1;
	for (i = 0; i < N; i++) {
		scanf("%d", &temp);
		if (flag&&!check(T, temp)) flag = 0;
	}
	if (flag) return true;
	else return false;
}
bool check(AVLTree T, int K) {
	if (!T->Flag) {
		if (K == T->Data) {
			T->Flag = 1;
			return true;
		}
		else return false;
	}
	else {
		if (K == T->Data) return false;
		if (K < T->Data) check(T->Left, K);
		else if (K > T->Data) check(T->Right, K);
	}
}
void Clear(AVLTree T) {
	if (!T) return;
	T->Flag = 0;
	Clear(T->Left);
	Clear(T->Right);
}

测试结果

问题整理

1.这次是第二次写这道题，很顺利地又出错了，上次忘记了BST的重置，这次忘记了TNode的Flag的重置。
2.判断工具的重置很重要，莫忘记。