PAT L3-010 是否完全二叉搜索樹
阿新 • • 發佈:2019-03-22
namespace .cn flag nbsp dfs tor -i highlight out
https://pintia.cn/problem-sets/994805046380707840/problems/994805049870368768
將一系列給定數字順序插入一個初始為空的二叉搜索樹(定義為左子樹鍵值大,右子樹鍵值小),你需要判斷最後的樹是否一棵完全二叉樹,並且給出其層序遍歷的結果。
輸入格式:
輸入第一行給出一個不超過20的正整數N
;第二行給出N
個互不相同的正整數,其間以空格分隔。
輸出格式:
將輸入的N
個正整數順序插入一個初始為空的二叉搜索樹。在第一行中輸出結果樹的層序遍歷結果,數字間以1個空格分隔,行的首尾不得有多余空格。第二行輸出YES
,如果該樹是完全二叉樹;否則輸出NO
。
輸入樣例1:
9
38 45 42 24 58 30 67 12 51
輸出樣例1:
38 45 24 58 42 30 12 67 51
YES
輸入樣例2:
8
38 24 12 45 58 67 42 51
輸出樣例2:
38 45 24 58 42 12 67 51
NO
代碼:
#include <bits/stdc++.h> using namespace std; const int maxn = 1e5 + 10; int N; int a[maxn]; vector<int> v(maxn); vector<int> ans[maxn]; int depth = -1, cnt = -1; struct Node{ int val; struct Node *left, *right; }; int Pow(int a, int b) { int ans = 1; if(b == 0) return 1; for(int i = 1; i <= b; i ++) ans *= a; return ans; } Node *BuildBST(Node *root, int x) { if(!root) { root = new Node(); root -> val = x; root -> left = NULL; root -> right = NULL; } else if(x <= root -> val) root -> right = BuildBST(root -> right, x); else root -> left = BuildBST(root -> left, x); return root; } void dfs(Node* root, int step, int index) { if(!root) { depth = max(depth, step + 1); return; } v[step] ++; ans[step].push_back(root -> val); dfs(root -> left, step + 1, index * 2); dfs(root -> right, step + 1, index * 2 + 1); cnt = max(cnt, index); } int height(Node* root) { if(!root) return 0; return max(height(root -> left), height(root -> right)) + 1; } int main() { scanf("%d", &N); Node *root = NULL; for(int i = 0; i < N; i ++) { scanf("%d", &a[i]); root = BuildBST(root, a[i]); } dfs(root, 0, 1); bool flag = true; for(int i = 0; i < depth - 2; i ++) { if(v[i] != Pow(2, i)) { flag = false; break; } } for(int i = 0; i < depth; i ++) { for(int j = 0; j < ans[i].size(); j ++) { if(i == 0 && j == 0) printf(""); else printf(" "); printf("%d", ans[i][j]); } } printf("\n"); if(height(root -> left) - height(root -> right) > 1) flag = false; if(cnt == N) printf("YES"); else printf("NO"); return 0; }
還有一種建樹一會寫吧
PAT L3-010 是否完全二叉搜索樹