给定一棵包含 N 个节点的完全二叉树,树上每个节点都有一个权值,按从上到下、从左到右的顺序依次是 A1,A2,⋅⋅⋅AN,如下图所示:
现在小明要把相同深度的节点的权值加在一起,他想知道哪个深度的节点权值之和最大?
如果有多个深度的权值和同为最大,请你输出其中最小的深度。
注:根的深度是 1。
输入格式
第一行包含一个整数 N。
第二行包含 N 个整数 A1,A2,⋅⋅⋅AN。
输出格式
输出一个整数代表答案。
数据范围
1≤N≤105,
−105≤Ai≤105
输入样例:
7
1 6 5 4 3 2 1
输出样例:
2
做法:1. 首先要知道完全二叉树是什么, 完全二叉树的定义是从上到下,从左到右,必须按顺序插入,即中间不能有空,也就是深度最深的一层可能都有结点,也可能不完全,但一定是从左到右,右边空着。
2. 根据这个性质可以推出每层的个数最多为2的k次方,然后枚举每层就可以了。其中C语言中的log2是用2为底的运算,sum要注意开long long.
#include <iostream>
#include <cstdio>
#include <cmath>
using namespace std;
#define ll long long
const int N = 1e6 + 10;
ll a[N];
int main(){
int n; scanf("%d", &n);
for(int i = 1; i <= n; i++){
scanf("%lld", &a[i]);
}
ll sum = 0, maxN = a[1];
int deep = 1;
for(int i = 1; i <= log2(n); i++){
sum = 0;
for(int j = ( 1 << i); j <= (1 << (i + 1)) - 1; j++){
sum += a[j];
}if(sum > maxN){
maxN = sum;
deep = i + 1;
}
}cout << deep;
return 0;
}