蓝桥-2019-完全二叉树的权值

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

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

注:根的深度是 1。

输入描述

第一行包含一个整数 N(1≤N≤1e5)。

第二行包含 N 个整数 A1​,A2​,⋅⋅⋅AN​(−1e5≤Ai​≤1e5)。

输出描述

输出一个整数代表答案。

输入输出样例

示例

输入

7
1 6 5 4 3 2 1

输出

2

我看到很多人题解都是什么pow(2,n)算来算去,我就不同了。这是一个完全二叉树,我们可以把它补成满二叉树,第一层节点数为1

第二层节点数为2

第三层为4......每次节点数都是2的次方。其实依次循环每个数组,只要是每次都是2的次方,那么就是一层。那么问题来了,如何判断是2的次方呢?这里我给大家一个很好的方法。

我们可以令m=pow(2,n),只要小于m并且能被m整除,那么肯定就是2的次方(),我在这个题目设了m=pow(2,30),此外,如何判断是3的次方,4的次方等等都能用类似的方法。前提是在你变量的范围内。

AC代码

#include <iostream>
#include <cmath>
#include <cstring>
#include <algorithm>
#include <climits>
using namespace std;
int main()
{
  int m=pow(2,30),n;//m用来测试是不是2的次方
  cin>>n;
  int ch[100010];
  int sum[1000],ans=-1,max=-INT_MAX,lmax=0;//先把max设计为最小。
  memset(sum,0,sizeof(sum));
  for(int i=1;i<=n;i++){
    cin>>ch[i];
    if(m%i==0){//如果i是2的次方,那么深度+1。因为这是完全二叉树
      ans++;
    }
    sum[ans]+=ch[i];
  }
  for(int i=0;i<=ans;i++){
    if(sum[i]>max){
      max=sum[i];
      lmax=i;
    }
  }
  cout<<lmax+1;//注意我们lmax是从0开始的,题目深度要求从1,+1即可。
  return 0;
}

总结:题目还是很简单的,当年我在力扣做这种题也都是pow一个一行算,有时候很容易分不清楚,所以我们开局直接用一个基准数来判断是不是2的次方即可(也是在力扣的大佬们学的),如果想练算法我真的推荐力扣,秀的你一脸。

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值