PTA 7-1 是否完全二叉搜索树 (30分)

PTA 7-1 是否完全二叉搜索树 (30分)

将一系列给定数字顺序插入一个初始为空的二叉搜索树（定义为左子树键值大，右子树键值小），你需要判断最后的树是否一棵完全二叉树，并且给出其层序遍历的结果。

输入格式：

输入第一行给出一个不超过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

【程序思路】

输入数据然后根据二叉搜索树的插入操作建树。
判断是否完全二叉树的要点： 我们按照层次顺序遍历这颗树的过程中，对于任意一节点x

• 如果x 有右子树，但是却没有左子树，这肯定不是完全二叉树
• （如果x 有左子树，但是却没有右子树）|| （ 如果 x 左右子树都没有）【归纳来说就是没有右子树】，那么剩余的所有节点一定要为叶子节点

【程序实现】

#include<bits/stdc++.h>
using namespace std;
typedef struct tree{
    int data;
    struct tree *left;
    struct tree *right;
}*Tree;
Tree Insert(Tree root, int x) {
    if (!root) {
        root = new struct tree;
        root->data = x;
        root->left = root->right = NULL;
    }
    else {
        if (x > root->data)
            root->left = Insert(root->left, x);
        else if (x < root->data)
            root->right = Insert(root->right, x);
    }
    return root;
}
bool VisitJdg(Tree root) {
    queue<Tree> q;
    int flag = 0;
    bool jdg = true;
    if (root) {
        q.push(root);
        while(!q.empty()) {
            Tree t = q.front();
            q.pop();
            cout<<t->data;
            if (t->left)
                q.push(t->left);
            if (t->right)
                q.push(t->right);
            if (!q.empty())
                cout<<' ';
            if (!flag) {
                if (!t->left && t->right)
                    jdg = false;
                else if (!t->right)
                    flag = 1;
            }
            else {
                if (t->left || t->right)
                    jdg = false;
            }
        }
        cout<<endl;
    }
    return jdg;
}
int main(){
    Tree root = NULL;
    int n, x;
    cin>>n;
    for (int i = 0; i < n; i++) {
        cin>>x;
        root = Insert(root, x);
    }
    if(VisitJdg(root))
        cout<<"YES\n";
    else
        cout<<"NO\n";
	return 0;
}