本题要点在于按照题目所给序列依次插入节点构造二叉树。构造二叉树采用l[x]和r[x]分别代表节点x的左右儿子的下标,若为0则说明该节点没有左/右儿子。
以1号节点为根节点,每次插入都从根节点开始,权值小于等于当前节点则前往左子树,否则前往右子树,直至找到一个空节点插入权值。最后dfs遍历一下所有节点,cnt记录下每一层的节点个数,输出最后两层即可。
需要注意的是insert函数中 int &u 实现了l[u]和r[u]的赋值。
#include <bits/stdc++.h>
using namespace std;
const int N = 1010;
int l[N], r[N], w[N];
int n, n1, n2;
int idx;
int cnt[N], maxDepth;
void insert(int &u, int weight)
{
if(!u)
{
u = ++idx;
w[u] = weight;
}
else if(weight <= w[u]) insert(l[u], weight);
else insert(r[u], weight);
}
void dfs(int u, int depth)
{
if(!u) return;
cnt[depth] ++;
maxDepth = max(depth, maxDepth);
dfs(l[u], depth + 1);
dfs(r[u], depth + 1);
}
int main()
{
cin >> n;
int root = 0;
while(n --)
{
int t;
cin >> t;
insert(root, t);
}
dfs(root, 0);
n1 = cnt[maxDepth], n2 = cnt[maxDepth - 1];
printf("%d + %d = %d", n1, n2, n1 + n2);
}