给定一棵包含 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的次方即可(也是在力扣的大佬们学的),如果想练算法我真的推荐力扣,秀的你一脸。