2019年第十届蓝桥杯A组E题完全二叉树的权值

参考博客

2019年第十届蓝桥杯【C++省赛B组】【第七题：完全二叉树的权值】——附解题思路及代码

 

【问题描述】
给定一棵包含 N 个节点的完全二叉树，树上每个节点都有一个权值，按从
上到下、从左到右的顺序依次是 A 1 , A 2 , ··· A N ，如下图所示：
现在小明要把相同深度的节点的权值加在一起，他想知道哪个深度的节点
权值之和最大？如果有多个深度的权值和同为最大，请你输出其中最小的深度。
注：根的深度是 1。
【输入格式】
第一行包含一个整数 N。
第二行包含 N 个整数 A 1 , A 2 , ··· A N 。
【输出格式】
输出一个整数代表答案。
【样例输入】
7
1 6 5 4 3 2 1

样例输出

2

对于每个节点，都计算该节点的深度，并将这个节点的权值，较到该深度上。 

#include<iostream>
#include<cstdio>
using namespace std;
const int MAX_DEEP=18;
long long temp[MAX_DEEP+1];
//获取下标为n的节点的深度 
int getDeep(int n)
{
	int res=0;
	while(n>0)
	{
		n=n/2;
		res++;
	}
	return res;
}
int main()
{
	int n,t;
	int deepweight[100000]={0};
	cin>>n;
	//输入n个节点。 
	for(int i=1;i<=n;i++)
	{
		cin>>t;
	    deepweight[getDeep(i)]=deepweight[getDeep(i)]+t;//第i个节点所在深度的权重和 deepweight[getDeep(i)] 
	}
	int maxweightsum=0;
	int index;
	for(int i=1;i<18;i++)
	{
		if(maxweightsum<deepweight[i])
		{
			maxweightsum=deepweight[i];
			index=i;
		}
	
	}
	cout<<index;	
 }