蓝桥杯真题：完全二叉树的权值

题目描述

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

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

注：根的深度是 1。

输入描述

第一行包含一个整数N（1≤N≤105）。

第二行包含 N 个整数A1​,A2​,⋅⋅⋅AN​（−105≤Ai​≤105）。

输出描述

输出一个整数代表答案。

输入输出样例

示例

输入

7
1 6 5 4 3 2 1

输出

2

运行限制

• 最大运行时间：1s
• 最大运行内存: 256M

直接暴力就完事：

#include <iostream>
#include <bits/stdc++.h>
using namespace std;


int main()
{
  // 请在此输入您的代码
  int N;
  cin>>N;
  int ans=INT_MIN,res=0;
  int layer=1,ans_layer=1;
  int num=1;
  for(int i=1;i<=N;++i)
  {
    if(num==0||i==N)
    {
      num=1<<(layer);
      if(res>ans) 
      {
        ans_layer=layer;
        ans=res;
      }
      if(i==N) break;
      layer++;
      res=0;
    }
    int a;
    cin>>a;
    res+=a;
    num--;
  }
  cout<<ans_layer<<endl;
  return 0;
}