蓝桥杯练习-3.8
• G 完全二叉树的权值
Description
给定一棵包含N个节点的完全二叉树,树上每个节点都有一个权值,按从
上到下、从左到右的顺序依次是A1,A2,· · ·AN,如下图所示:
现在小明要把相同深度的节点的权值加在一起,他想知道哪个深度的节点
权值之和最大?如果有多个深度的权值和同为最大,请你输出其中最小的深度。
注:根的深度是1。
Input
第一行包含一个整数N。
第二行包含N个整数A1,A2,· · ·AN
Output
输出一个整数代表答案。
Sample Input 1
7
1 6 5 4 3 2 1
Sample Output 1
2
Hint
对于所有评测用例,1≤N≤100000,100000≤A_i≤1000001≤N≤100000,100000≤A**i≤100000。
思路
需要了解二叉树的性质
• 在二叉树的第i层上最多有2^(i-1) 个节点 。(i>=1)
• 二叉树中如果深度为k,那么最多有2^k - 1个节点。(k>=1)
• 在完全二叉树中,具有n个节点的完全二叉树的深度为[log2n]+1,其中[log2n]是向下取整。
我们只需要把每一深度的结点的权值加在一起,找到最大的那个即可。
因为测试用例 N ≤ 100000,而2^17 = 131072,所以深度最多k最多为17,而且17层并未填满,也需要判断,是否不是满二叉树的情况。因为最后一层可能并没有填满。
#include <iostream>
#include <cmath>
#include <algorithm>
using namespace std;
long long ans=-10000000005;//权值总和
int d;//记录最终深度
int a[100005];//输入的每个结点的权值
int main()
{
int n;//输入多少个结点的权值
int depth;//计算最大有多深
cin >> n;
for (int i = 1; i <= n; i++)
{
cin >> a[i];
}
for (int i = 1; i <= 17; i++)//计算出最大的深度
{
if (n >= pow(2,i - 1) && n <= pow(2,i) - 1)
{
depth = i;
break;
}
}
for (int i = 1; i <= depth; i++)//开始循环每一层树了
{
long long sum = 0;//每次都要初始化记录权值的变量
for (int j = pow(2, i - 1); j <= pow(2, i) - 1 && j <= n; j++)//在第i层的开始,第一个是第2^(i-1)-1+1个,最后一个是2^k - 1
{
sum += a[j];
}
if (sum == ans)//如果有一样大的,取深度最小的那个
{
d = min(d, i);
}
if (sum > ans)
{
ans = sum;
d = i;
}
}
cout << d << endl;
return 0;
}