完全二叉树的权值

题目描述

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

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

输入

第一行包含一个整数 N。 第二行包含N个整数A1,A2,··· AN。

输出

输出一个整数代表答案。

样例输入

7
1 6 5 4 3 2 1

样例输出

2

代码：

#include <iostream>
#include <algorithm>
using namespace std;
int s[1005],a[1005];
bool cmp(int a,int b)
{
    return a>b;
}
int main()
{
    int n,num;
    cin>>n;
    int k=1,flag=0,temp=1;
    /* k 数组的下标变量,即每个深度的权值
         都被加和后放在数组s的同一下标内*/
    /* flag 表示已被放入数组s内的个数
              temp表示每个深度的节点个数*/
    while(n--)
    {
        cin>>num;
        if(flag<temp)
        {
            s[k]+=num;
            a[k]+=num;
            /*另定义了一个与数组s相同的数组a
              便于找出最大权和的（最小）深度*/
            flag++;
        }

        if(flag==temp)
        {
            flag=0;
            temp*=2;
            k++;
        }
    }
    sort(a,a+1005,cmp);
    /*对数组a进行排序，找出最大值*/
    int ans;
    for(int i=1;;i++)
    {
        if(s[i]==a[0])
        {
            ans=i;
            break;
        }
    }
    cout<<ans;
    return 0;
}

• 这个是我自己写的，能过的数据样例应该很少。下附一份学长推荐的博主的AC代码，点此跳转。

#include <bits/stdc++.h>
using namespace std;
const int MAXN = 1000000;
int N, a[MAXN + 1];
int main() {
    scanf("%d", &N);
    for (int i = 1; i <= N; ++i)
        scanf("%d", a + i);
    long long m = -MAXN;
    int ans = 0;
    for (int i = 1; (1 << (i - 1)) < N; ++i) {
        long long t = 0;
        for (int j = 1 << (i - 1); j < (1 << i) && j < N; ++j)
            t += a[j];
        if (m < t) {
            m = t;
            ans = i;
        }
    }
    cout << ans << endl;
    return 0;
}

总结

• 虽然过了题，可能是因为数据样题不全。
• 第二份代码看不懂，还是得继续加油鸭！