L2-3 二叉搜索树的2层结点统计
L2-3 二叉搜索树的2层结点统计
分数 25
作者 陈越
单位 浙江大学
二叉搜索树或者是一棵空树,或者是具有下列性质的二叉树:若它的左子树不空,则左子树上所有结点的值均小于或等于它的根结点的值;若它的右子树不空,则右子树上所有结点的值均大于它的根结点的值;它的左、右子树也分别为二叉搜索树。
将一系列数字按给定顺序插入一棵初始为空的二叉搜索树,你的任务是统计结果树中最下面 2 层的结点数。
输入格式:输入在第一行给出一个正整数 N (≤1000),为插入数字的个数。第二行给出 N 个 [−1000,1000] 区间内的整数。数字间以空格分隔。
输出格式:在一行中输出最下面 2 层的结点总数。
输入样例:9
25 30 42 16 20 20 35 -5 28
输出样例:6
本来想用静态数组,结果感觉最大要2^1000,空间过不了,所以用指针
#include<iostream>
#include<vector>
#include<algorithm>
#include<math.h>
#include<sstream>
#include<string>
#include<string.h>
#include<iomanip>
#include<map>
#include<queue>
#include<limits.h>
#include<climits>
using namespace std;
int n;
struct tree
{
tree* left;
tree* right;
int val;
int height;
tree() { left = NULL; right = NULL; val = 0; }
};
tree* lst;
int ans = 0;
void dfs(tree* t,int man)
{
if (t->height > man - 2 and t->height <= man)ans++;
if (t->right != NULL)dfs(t->right, man);
if (t->left != NULL)dfs(t->left, man);
}
int main()
{
lst = new tree;
int n;
cin >> n;
int xx;
cin >> xx;
lst->val = xx;
lst->height = 1;
tree* start = lst;
int heightmax = 1;
for (int i = 1; i < n; i++)
{
start = lst;
tree *next = new tree;
cin >> xx;
next->val = xx;
while ( start->height <= heightmax)
{
if (next->val > start->val)
{
if (start->right != NULL)start = start->right;
else break;
}
else
{
if (start->left != NULL)start = start->left;
else break;
}
}
if (next->val > start->val)
{
start->right = next;
}
else
{
start->left=next;
}
next->height = start->height + 1;
heightmax = max(heightmax, next->height);
}
start = lst;
dfs(start, heightmax);
cout << ans << endl;
return 0;
}