04-樹4是否同一棵二叉搜尋樹(25分)

給定一個插入序列就可以唯一確定一棵二叉搜尋樹。然而，一棵給定的二叉搜尋樹卻可以由多種不同的插入序列得到。例如分別按照序列{2, 1, 3}和{2, 3, 1}插入初始為空的二叉搜尋樹，都得到一樣的結果。於是對於輸入的各種插入序列，你需要判斷它們是否能生成一樣的二叉搜尋樹。

輸入格式:

輸入包含若干組測試資料。每組資料的第1行給出兩個正整數N (≤)和L，分別是每個序列插入元素的個數和需要檢查的序列個數。第2行給出N個以空格分隔的正整數，作為初始插入序列。最後L行，每行給出N個插入的元素，屬於L個需要檢查的序列。

簡單起見，我們保證每個插入序列都是1到N的一個排列。當讀到N為0時，標誌輸入結束，這組資料不要處理。

輸出格式:

對每一組需要檢查的序列，如果其生成的二叉搜尋樹跟對應的初始序列生成的一樣，輸出“Yes”，否則輸出“No”。

輸入樣例:

4 2
3 1 4 2
3 4 1 2
3 2 4 1
2 1
2 1
1 2
0

輸出樣例:

Yes
No
No

提交程式碼：

#include <stdio.h>
#include <stdlib.h>

typedef struct TreeNode* Tree;
struct TreeNode{
    int v;
    Tree left;
    Tree right;
    
 
int flag;
};

//構建新的結點
Tree NewNode(int V){
    Tree T = (Tree)malloc(sizeof(struct TreeNode));
    T->v = V;
    T->left = NULL;
    T->right = NULL;
    T->flag = 0;
    return T;
}
//插入結點
Tree Insert(Tree T, int V){
    if(!T) T = NewNode(V);
    else{
        if(V > T->v)
            T
 
->right = Insert(T->right, V);
        else
            T->left = Insert(T->left, V);
    }
    return T;
}

//構建樹
Tree MakeTree(int N){
    Tree T;
    int i, V;
    scanf("%d", &V);
    T = NewNode(V);
    for(i = 1; i < N; i++){
        scanf("%d", &V);
        T = Insert(T, V);
    }
    return T;
}

int check(Tree T, int V){
    if(T->flag){
        if(V < T->v) return check(T->left, V);
        else if(V > T->v) return check(T->right, V);
        else return 0;
    }
    else{
        if(V==T->v){
            T->flag = 1;
            return 1;
        }
        else return 0;
    }
}


int JudgeTree(Tree T, int N){
    int i, V;
    int flag = 1; //值為1時，樹相同；值為0時，樹不相同；
    scanf("%d", &V);
    if(V!=T->v) flag = 0;
    else T->flag = 1;
    
    for(i = 1; i < N; i++){
        scanf("%d", &V);
        if(flag && !check(T, V)) flag = 0;
    }
    return flag;
}

void ResetTree(Tree T){
    if(T->left) ResetTree(T->left);
    if(T->right) ResetTree(T->right);
    T->flag = 0;
}

void FreeTree(Tree T){
    if(T->left) FreeTree(T->left);
    if(T->right) FreeTree(T->right);
    free(T);
}

int main(){
    //Input data.
    int N,L;
    Tree T;
    scanf("%d", &N);
    while(N){
        scanf("%d", &L);
        T = MakeTree(N);
        for(int i = 0; i < L; i++){
            if(JudgeTree(T, N))printf("Yes\n");
            else printf("No\n");
            ResetTree(T);
        }
        FreeTree(T);
        scanf("%d", &N);
    }
    return 0;
}

測試結果：