蓝桥杯--完全二叉树的权值【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;
}