1. 程式人生 > >PTA 7-1 是否同一棵二叉搜尋樹(25 分) 建樹比較

PTA 7-1 是否同一棵二叉搜尋樹(25 分) 建樹比較

利用二叉樹性質

建樹,比較

#include<bits/stdc++.h>

using namespace std;
const int maxn = 1024 + 7;

int n, m;
int a[maxn], b[maxn];

void build1() {
    memset(a, -1, sizeof a);
    for(int i = 0; i < n; ++i) {
        int id = 1, x;
        scanf("%d", &x);
        while(1) {
            if(a[id] == -1) {
                a[id] = x; break;
            }
            else if(x < a[id]) {
                id *= 2;
            }
            else id = 2*id+1;
        }
    }
}

void build2() {
    memset(b, -1, sizeof b);
    for(int i = 0; i < n; ++i) {
        int id = 1, x;
        scanf("%d", &x);
        while(1) {
            if(b[id] == -1) {
                b[id] = x; break;
            }
            else if(x < b[id]) {
                id *= 2;
            }
            else id = 2*id+1;
        }
    }
}

bool check() {
    for(int i = 1; i < maxn; ++i) {
        if(a[i] != b[i]) return false;
    }
    return true;
}

int main() {

    while(~scanf("%d %d", &n, &m) && n>0) {
        build1();
        for(int i = 0; i < m; ++i) {
            build2();
            if(check()) puts("Yes");
            else puts("No");
        }
    }

    return 0;
}