完全二叉树的权值
题目描述
核心思路
代码
#include<iostream>
using namespace std;
typedef long long LL;
const int N=100010;
int n;
int a[N];
int main()
{
scanf("%d",&n);
for(int i=1;i<=n;i++)
scanf("%d",&a[i]);
int depth=-0; //最小的深度
LL maxs=-1e18;
//d控制层数 i是数组下标
for(int d=1,i=1;i<=n;i*=2,d++)
{
LL s=0;//统计某一层内所有结点的权值之和
for(int j=i;j<i+(1<<d-1)&&j<=n;j++)
{
s+=a[j];
}
//找到权值之和最大的那一层
if(s>maxs)
{
maxs=s;
depth=d;
}
}
printf("%d\n",depth);
return 0;
}