1. 程式人生 > >PAT L3-010 是否完全二叉搜索樹

PAT L3-010 是否完全二叉搜索樹

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 是否完全二叉搜索樹