7-4 是否同一棵二叉搜尋樹(25 分)
阿新 • • 發佈:2018-12-24
給定一個插入序列就可以唯一確定一棵二叉搜尋樹。然而,一棵給定的二叉搜尋樹卻可以由多種不同的插入序列得到。例如分別按照序列{2, 1, 3}和{2, 3, 1}插入初始為空的二叉搜尋樹,都得到一樣的結果。於是對於輸入的各種插入序列,你需要判斷它們是否能生成一樣的二叉搜尋樹。
輸入格式:
輸入包含若干組測試資料。每組資料的第1行給出兩個正整數N (≤10)和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
tip:利用陣列建樹來進行比較;
AC程式碼:
#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; } } } int check() { for(int i = 1; i < maxn; ++i) { if(a[i] != b[i]) return 0; } return 1; } int main() { while(cin>>n>>m) { if(n==0) break; build1(); for(int i = 0; i < m; ++i) { build2(); if(check()) cout<<"Yes"; else cout<<"No"; } } return 0; }