P8681 [蓝桥杯 2019 省 AB] 完全二叉树的权值

题目描述

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

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

        注：根的深度是 1。

输入格式

        第一行包含一个整数 N。

        第二行包含 N 个整数 A1,A2, …,AN。

输出格式

        输出一个整数代表答案。

样例 1

样例输入 1

7
1 6 5 4 3 2 1

样例输出 1

2

提示

        对于所有评测用例，1 ≤ N ≤ 10^5，0 ≤ |Ai| ≤ 10^5。

蓝桥杯 2019 省赛 A 组 F 题（B 组 G 题）。

代码

#include <iostream>
#include <algorithm>
#include <cmath>
#include <limits.h>
using namespace std;

int main()
{
	int n;
	cin>>n;
	int a[n+1];
	for(int i = 1; i <= n; i++)
	{
		cin>>a[i];
	}
	int k = n;
	int m = 0;
	while(n != 0)
	{
		n /= 2;
		m++;
	}
	int max = INT_MIN;
	int layer;
	for(int i = 0; i < m; i++)
	{
		int sum = 0;
		for(int j = pow(2,i); j < pow(2,i+1) && j <= k; j++)
		{
			sum += a[j];
		}
		if(sum > max)
		{
			max = sum;
			layer = i + 1;
		}
	}
	cout<<layer;
	return 0;
}