蓝桥杯--完全二叉树的权值【2019第十届蓝桥杯C/C++省赛B组】

蓝桥杯--完全二叉树的权值【2019第十届蓝桥杯C/C++省赛B组】

题目描述

给定一棵包含 N 个节点的完全二叉树，树上每个节点都有一个权值，按从 上到下、从左到右的顺序依次是 A1, A2, ··· AN。

现在小明要把相同深度的节点的权值加在一起，他想知道哪个深度的节点 权值之和最大？如果有多个深度的权值和同为最大，请你输出其中最小的深度。

注：根的深度是 1。

输入描述

第一行包含一个整数 N（1 <= N <= 10^5）。

第二行包含 N 个整数 A1, A2, ··· AN （−10^5 <= Ai <= 10^5）。

输出描述

输出一个整数代表答案。

输入输出样例

示例

输入

7
1 6 5 4 3 2 1

输出

2

运行限制

• 最大运行时间：1s
• 最大运行内存: 256M

思路

利用完全二叉树特点判断哪些结点属于同一行，并进行求和，不断更新最大值深度即可。

代码

#include<iostream>
#include<cmath>

using namespace std;

int deep=1;
int max_deep;
int max_sum;
int sum;
int n;
int num;
double ans;

int main()
{
	cin>>n;
	while(n--)
	{
		ans++;
		cin>>num;
		sum+=num;
		if(ans==pow(2,deep)-1)
		{
			if(sum>max_sum)
			{
				max_sum=sum;
				max_deep=deep;
			}
			deep++;
			sum=0;
		}
	}
	if(ans!=pow(2,deep-1)-1)
	{
		if(sum>max_sum)
			max_deep=deep;
	}
	cout<<max_deep;
	return 0;
}