题目地址:
https://www.acwing.com/problem/content/1242/
给定一棵包含 N N N个节点的完全二叉树,树上每个节点都有一个权值,按从上到下、从左到右的顺序依次是 A 1 , A 2 , ⋅ ⋅ ⋅ , A N A_1,A_2,···,A_N A1,A2,⋅⋅⋅,AN。现在小明要把相同深度的节点的权值加在一起,他想知道哪个深度的节点权值之和最大?如果有多个深度的权值和同为最大,请你输出其中最小的深度。注:根的深度是 1 1 1。
输入格式:
第一行包含一个整数
N
N
N。第二行包含
N
N
N个整数
A
1
,
A
2
,
⋅
⋅
⋅
,
A
N
A_1,A_2,···,A_N
A1,A2,⋅⋅⋅,AN。
输出格式:
输出一个整数代表答案。
数据范围:
1
≤
N
≤
1
0
5
1≤N≤10^5
1≤N≤105
−
1
0
5
≤
A
i
≤
1
0
5
−10^5≤A_i≤10^5
−105≤Ai≤105
第 k k k层的节点的编号是从 2 k − 1 2^{k-1} 2k−1到 2 k − 1 2^k-1 2k−1。枚举每一层即可。代码如下:
#include <iostream>
#include <climits>
using namespace std;
const int N = 1e5 + 10;
int n;
long a[N];
int main() {
scanf("%d", &n);
for (int i = 1; i <= n; i++) scanf("%ld", &a[i]);
int res = 0;
long sum = 0, maxs = LONG_MIN;
for (int k = 1; k <= n; k++) {
sum = 0;
int i = 1;
for (i = 1 << k - 1; i < min((long) n + 1, 1L << k); i++)
sum += a[i];
if (sum > maxs) {
res = k;
maxs = sum;
}
if (i == n + 1) break;
}
printf("%ld\n", res);
return 0;
}
时空复杂度 O ( N ) O(N) O(N)。