完全二叉树的权值

最新推荐文章于 2023-03-14 16:03:10 发布

WananRd

最新推荐文章于 2023-03-14 16:03:10 发布

阅读量660

点赞数

分类专栏： # 蓝桥杯

本文链接：https://blog.csdn.net/weixin_51173281/article/details/115273063

版权

完全二叉树节点权值深度最大和算法

关键词由CSDN通过智能技术生成

蓝桥杯专栏收录该内容

81 篇文章 14 订阅

订阅专栏

题目描述
给定一棵包含N 个节点的完全二叉树，树上每个节点都有一个权值，按从上到下、从左到右的顺序依次是A1, A2, AN，如下图所示：
在这里插入图片描述
现在小明要把相同深度的节点的权值加在一起，他想知道哪个深度的节点权值之和最大？如果有多个深度的权值和同为最大，请你输出其中最小的度。
注：根的深度是1。

输入
第一行包含一个整数N。
第二行包含N 个整数A1, A2, AN
对于所有评测用例，1<=N<=100000, -100000<=Ai<=100000。

输出
输出一个整数代表答案。

样例输入 Copy
7
1 6 5 4 3 2 1

样例输出 Copy
2

思路：将每一层的sum存放在数组中。

代码：

#include<iostream>
#include<cmath>
using namespace std;
int main()
{
	int a[100010] = { 0 };
	int n, num;
	cin >> n;
	for (int i = 1; i <= n; i++)
	{
		cin >> num;
		a[(int)floor(log2(i))+1] += num;
	}
	int max = a[1];
	int index = 1;
	for (int i = 1; i <= floor(log2(n))+1; i++)
	{
		if (a[i] > max)
		{
			max = a[i];
			index = i;
		}
	}
	cout << index;
	return 0;
}