1115 Counting Nodes in a BST （30 分）

这道题我最开始题目理解错了，最开始我理解成了统计叶子节点和叶子节点的上一层的结点数量。一提交得了2分，这时我知道我题目肯定读错了。
后来又做了一次，然后提交得了8分，我实在想不出有什么地方没写对，但是就是通不过。
后来认真读了一遍题目，才发现这个二叉搜索树种有重复数值的结点，而且要按照题目，把小于等于根结点的数据放在左子树，我想当然的放在了右子树，所以导致了错误，后来把这个错误改正过来了，一提交，果然通过了。
这个故事告诉我们认真读题是多么的重要。
后来翻看别人博客的时候，看到有个很简洁的写法，所以就办了过来。在插入的时候，同时记录结点的层数。

简洁版

#include<cstdio>
#include<vector>
#include<algorithm>
using namespace std;
struct node{
	int data;
	node *left,*right;
	node(int x): data(x),left(NULL),right(NULL){} //构造函数 
};
int maxL=0;
vector<int> level(1005,0); //初始化 
void insert(node* &root, int x,int depth){
	if(root == NULL){
		root = new node(x);
		level[depth]++; 
		maxL = max(depth, maxL); //之前这个地方写错了 ，这里应该取深度的最大值，而不是深度出现次数的最大值 
	}
	else if(x > root->data ){
		insert(root->right, x,depth+1);
	}else{
		insert(root->left, x,depth+1);
	}
} 


int main(){
	int n,data;
	node* root = NULL;
	scanf("%d",&n);
	for(int i=0;i<n;i++){
		scanf("%d",&data);
		insert(root, data,0);
	}
	printf("%d + %d = %d\n",level[maxL],level[maxL-1],level[maxL] + level[maxL-1]);
	return 0;
}

笨比版

#include<cstdio>
int n1,n2;
struct node{
	int data,depth;
	node *left,*right;
};
void insert(node* &root, int x){
	if(root == NULL){
		root = new node;
		root->data = x;
		root->left = root->right = NULL; 
	}
	else if(x > root->data ){
		insert(root->right, x);
	}else{
		insert(root->left, x);
	}
} 
int max=0;
void find(node* root){
	if(root->left != NULL){
		root->left->depth = root->depth +1;
		if(root->left->depth > max) max = root->left->depth;
		find(root->left);
	}
	if(root->right != NULL){
		root->right->depth = root->depth +1;
		if(root->right->depth > max) max = root->right->depth;
		find(root->right);
	}
}
void preorder(node* root){
	if(root == NULL) return;
	if(root->depth == max-1) n2++;
	else if(root->depth == max) n1++;
	preorder(root->left);
	preorder(root->right);
}
int main(){
	int n,data;
	node* root = NULL;
	scanf("%d",&n);
	for(int i=0;i<n;i++){
		scanf("%d",&data);
		insert(root, data);
	}
	root->depth = 0;
	find(root);
	preorder(root);
	printf("%d + %d = %d\n",n1,n2,n1+n2);
	return 0;
}

亲为版

#include <cstdio>
#include <iostream>
#include <vector>
#include <algorithm>
#include <queue>
using namespace std;
int n;
struct node
{
    int val;
    node *left, *right;
    node(int x)
    {
        val = x;
        left = right = nullptr;
    }
};
void insert(node *&root, int x)
{
    if (!root)
        root = new node(x);
    else if (x <= root->val)
    {
        insert(root->left, x);
    }
    else
        insert(root->right, x);
}
vector<int> level;
void levelorder(node *root)
{
    queue<node *> q;
    q.push(root);
    while (q.size())
    {
        int len = q.size();
        level.push_back(len);
        
        for (int j = 0; j < len; j++)
        {
            node* t = q.front();
            q.pop();
            if (t->left)
                q.push(t->left);
            if (t->right)
                q.push(t->right);
        }
        
    }
}

int main()
{
    scanf("%d", &n);
    int x;
    node *root = nullptr;
    for (int i = 0; i < n; i++)
    {
        scanf("%d", &x);
        insert(root, x);
    }
 	levelorder(root);
 	int m = level.size();
    printf("%d + %d = %d\n", level[m - 1], level[m - 2], level[m - 1] + level[m - 2]);
    return 0;
}