蓝桥杯2019真题 完全二叉树的权值

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

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

注：根的深度是 1。

输入描述

第一行包含一个整数 N（1 \leq N \leq 10^5）N（1≤N≤105）。

第二行包含 N 个整数 A_1, A_2, ··· A_N （−10^5 \leq A_i \leq 10^5）。

输出描述

输出一个整数代表答案。

输入输出样例

7
1 6 5 4 3 2 1

2

运行限制

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

 代码：

#include <iostream>
#include <cmath>
#include <algorithm>
using namespace std;

//完全二叉树可以用数组存储
int main() {
	int num;
	cin >> num;
    
    //C++没有以2为底的求解函数，但又以10为底的公式，可以利用换底公式求解
    //得到完全二叉树的高度
	int N = int(ceil(log10(num) / log10(2)));   //ceil向上取整，floor向下取整

	int* a = new int[num];//新建一个数组存储各个结点的权重
	int* sum = new int[N];//新建一个数组存储相同深度结点权重之和
	
	for (int j = 0; j < N; j++) {
		sum[j] = 0;
	}

	int tmp = 0;
	for (int i = 1; i <= num; i++) { //i从1开始，方便求对数
		cin >> a[i-1];
     
		tmp = int(floor(log10(i) / log10(2))); //完全二叉树同一深度的结点，求得的tmp的值是相同的
		sum[tmp] += a[i-1];
	}
    
    //库函数max_element找到sum中最大的元素，该函数的返回值是地址，减去起始地址即为下标
	int max_indx = max_element(sum, sum + N) - sum;
    
    //根结点深度为1，结果需要增加1
	cout << max_indx + 1;
}