2.8 【完全二叉树的权值】

题目描述

输入描述

输出描述

输出一个整数代表答案

输入输出样例

输入：

7
1 6 5 4 3 2 1

输出：

2

---

最终代码

1. c/c++

#include<bits/stdc++.h>
using namespace std;
int a[101000];

//代表2^x
int p(int x)
{
    int t=1;
    while(x--) t*=2;
     return t;
}
int main()
{
    //n是节点个数，t代表"权值和最大"的最小层
    //i代表节点是第几个，j代表层数
    int n,t=1,i=2,j=2;

    scanf("%d %d",&n,&a[1]);

    long long maxx=a[1];
    for(int i=2;i<=n;i++) scanf("%d",&a[i]);

    while(i<=n)
    {
        long long k=0;
        for(;i<p(j)&&i<=n;i++) k+=a[i];

        if(k>maxx)
        {
            maxx=k;
            t=j;
        }

        j++;
    }


   cout<<t<<endl;
   return 0;
}

---

2. java

import java.util.Scanner;
 
public class Main {
    static long sum[] = new long[100];
     
    public static void main(String[] args) {
        Scanner in = new Scanner(System.in);
        int n = in.nextInt();
        int m = (int) Math.floor(Math.log(n)/Math.log(2));
        for(int i=1;i<=n;i++) {
            long temp = in.nextLong();
            sum[(int) Math.floor(Math.log(i)/Math.log(2))] += temp;
        }
        long max = sum[0];
        int ans = 1;
        for(int i=1;i<=m;i++) {
            if(sum[i]>max) {
                max = sum[i];
                ans = i + 1;
            }
        }
        System.out.println(ans);        
    } 
}

---

3. python

p = []
for i in range(0, 33):  #32层。
    p.append(2 ** i)    #每层的个数：p[]=1,2,4,8,16,...
 
n = int(input())
a = input().split(" ")
 
ans = [0] * 32  #记录每层的和

for i in range(0, n):
    for j in range(0, len(p)):
        if i + 1 >= p[j] and i + 1 < p[j + 1]: #计算每层的和
            ans[j] += int(a[i])
 
mx, u = -1, -1
for i in range(0, len(ans)):
    if mx < ans[i]:
        mx, u = ans[i], i
print(u + 1)

---

过程理解