蓝桥杯 完全二叉树的权值

【样例输入】

7
1 6 5 4 3 2 1

【样例输出】

2

一棵深度为k且有 2k -1个结点的二叉树称为满二叉树。
一棵深度为k的有n个结点的二叉树，对树中的结点按从上至下、从左到右的顺序进行编号，如果编号为i（1≤i≤n）的结点与满二叉树中编号为i的结点在二叉树中的位置相同，则这棵二叉树称为完全二叉树。满二叉树是特殊的完全二叉树。

法一：输入与数据处理同步

#include<iostream>
#include<math.h>
using namespace std;
/*
1、判断层数
2、依次求层次 权值和
*/

//求层次
int comLayer(int x) {
	int layer = 0;
	int sum = 0;
	while (sum < x) {
		sum += pow(2, layer);
		layer++;
	}
	return layer;
}

int main() {
	int n, sum=0, count = 0;
	int k = 0;
	int layer, max_layer = 0;//层数
	scanf_s("%d", &n);
	layer = comLayer(n);//计算层数
	for (int i = 0; i < layer; i++) {
		k += count;
		count = pow(2, i);//第i+1层的个数
		int temp, tsum;
		for (int j = 0; j < count && j < n-k; j++) {
			tsum=0;
			cin >> temp;
			tsum += temp;
		}
		//更新权值最大和
		if (tsum > sum) {
			sum = tsum;
			max_layer = i+1;
		}
	}
	cout << max_layer;
	return 0;
}

法二：先输入后数据处理

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

int a[1001000];

int main()
{
    int ans=0;
    int n;
    cin>>n;
    
    for(int i=1;i<=n;i++){
        cin>>a[i];
    }
    int i=1,max=0,j=1;//i表示节点编号，j表示层数
    while(i<=n){
        int s=0;
        for(;i<=pow(2,j)-1;i++) s+=a[i];//计算j层所有节点的和
        if(s>max){
            max=s;
            ans=j;
        }
        j++;//下一层
    }
    cout<<ans;
    
    return 0;
}

个人人为此题能够加深对for、while循环的掌握。