PAT A1115 Counting Nodes in a BST (30 分) 二叉搜索树

    题目大意：给出一个序列，按照序列的顺序来构造BST，注意这里BST的定义中，左子树是不大于父亲节点的，右子树是大于父亲节点的。然后输出最低的两层的节点总数。

     首先按照序列顺序构造BST。因为要输出最低两层的节点总数，自然想到层序遍历，并且给节点加上记录层数的域layer。然后在层序遍历的同时更新节点的layer域的值，以及最大的层数maxLevel。

      为了避免得到最大层数之后再遍历一遍，开一个长度与序列长度相等的数组 table ，在层序遍历的同时记录每一层的节点数，因为最大的层数也就是序列长度了。最后只要输出 table[maxLevel] + table[maxLevel-1] 的值就好了。

AC代码如下：

#include <iostream>
#include <vector>
#include <algorithm>
#include <cstdio>
#include <cmath>
#include <stack>
#include <queue>
#include <set>
#include <map>
#include <cstring>

using namespace std;

struct Node
{
    int value;
    Node* left;
    Node* right;
    int layer;
    Node(int value)
    {
        this->value = value;
        this->layer = 0;
        this->right = this->left = NULL;
    }
};

void insert(Node* &node, int value)
{
    if(node == NULL)
    {
        node = new Node(value);
        return;
    }
    if(value <= node->value) insert(node->left, value);
    else insert(node->right, value);
}

void levelOrder(Node* root, vector<int> &table, int &maxLevel)
{
    queue<Node*> q;
    q.push(root);
    while(!q.empty())
    {
        Node* front = q.front();
        q.pop();
        table[front->layer]++;
        if(front->layer > maxLevel) maxLevel = front->layer;
        if(front->left)
        {
            front->left->layer = front->layer + 1;
            q.push(front->left);
        }
        if(front->right)
        {
            front->right->layer = front->layer + 1;
            q.push(front->right);
        }
    }
}

int main()
{
    Node* root = NULL;
    int N;
    cin >> N;
    for (int i = 0; i < N; ++i)
    {
        int value;
        scanf("%d", &value);
        insert(root, value);
    }
    vector<int> levelCount(N);
    int maxLevel = 0;
    levelOrder(root, levelCount, maxLevel);
    printf("%d + %d = %d", levelCount[maxLevel], levelCount[maxLevel-1], levelCount[maxLevel] + levelCount[maxLevel-1]);
    return 0;
}