蓝桥杯2019年第十届真题-完全二叉树的权值 （C++代码）

输入的时候计算每层的和；

#include<iostream>
#define lld long long int
using namespace std;
int tree(int n) {
    int x = 1, y = 0, c;//
    lld max=0;
    while(n>0) {
        lld sum=0;
        //完全二叉树最后一层可能不满，所以 i 要小于 min(n ,x);
        for(int i = 0; i < min(n,x); i++) {
            int t;
            cin >> t;
            //cout << t << " ";//输出
            sum+=t;
        }
        //cout << endl;
        y++;
        if(max < sum) {
            max = sum;
            c = y;
        }
        n = n - x;
        x = x * 2;
    }
    return c;
}
int main() {
    int n;
    while(cin >> n) {
        cout << tree(n) << endl;
    }
    return 0;
}