题目描述
给定一棵包含N 个节点的完全二叉树,树上每个节点都有一个权值,按从
上到下、从左到右的顺序依次是A1, A2, AN,如下图所示:
现在小明要把相同深度的节点的权值加在一起,他想知道哪个深度的节点
权值之和最大?如果有多个深度的权值和同为最大,请你输出其中最小的深度。
注:根的深度是1。
输入
第一行包含一个整数N。
第二行包含N 个整数A1, A2, AN
对于所有评测用例,1<=N<=100000, -100000<=Ai<=100000。
输出
输出一个整数代表答案。
样例输入
7
1 6 5 4 3 2 1
样例输出
2
【题解】1.二叉树性质是解题的关键,不记得也没有关系,可以找个规律什么的。节点的序号就决定了它在第几层,输入的时候去加。最后比较一下就可以了。
一般求层数的正常求法是log2(n),但是如果用c语言自带的log,会有取整等一系列的问题。
2.所以自己写一个求log2 的函数,计算这个数能除几次2。
3.记得开long long
代码
#include <bits/stdc++.h>
#define ll long long
using namespace std;
const int maxn=100005;
ll a[maxn];
ll sum[100];
int lg(int x)
{
int cnt=-1;
while(x)
{
x>>=1;
cnt++;
}
return cnt;
}
int main()
{
int n;
scanf("%d",&n);
int m=lg(n);
for(int i=1;i<=n;i++)
{
scanf("%lld",&a[i]);
sum[lg(i)]+=a[i];
}
ll maxx=sum[0];
int ans=1;
for(int i=1;i<=m;i++)
{
if(sum[i]>maxx)
{
maxx=sum[i];
ans=i+1;
}
}
cout<<ans<<endl;
}