河南省蓝桥杯第十届省赛-6-完全二叉树的权值

题目描述:

【问题描述】 给定一棵包含 N 个节点的完全二叉树，树上每个节点都有一个权值，按从 上到下、从左到右的顺序依次是 A1, A2, · · · AN，如下图所示： 现在小明要把相同深度的节点的权值加在一起，他想知道哪个深度的节点 权值之和最大？如果有多个深度的权值和同为最大，请你输出其中最小的深度。 注：根的深度是 1。

【输入格式】 第一行包含一个整数 N。 第二行包含 N 个整数 A1, A2, · · · AN 。

【输出格式】 输出一个整数代表答案。 【样例输入】 7 1 6 5 4 3 2 1

【样例输出】 2

【评测用例规模与约定】 对于所有评测用例，1 ≤ N ≤ 100000，−100000 ≤ Ai ≤ 100000

分析:

很简单,很显然是用BFS来算

代码如下:

#include<iostream>
#include<math.h>
using namespace std;
struct node{
	int num=-1;
	int weight;
	int level;
};
node nodelist[100005];
int n;
int callevel(int k);
int main()
{
	cin>>n;
	for(int i=1;i<=n;i++)
	{
		cin>>nodelist[i].weight;
		nodelist[i].num=i;
		nodelist[i].level=callevel(i);
	}
	/*
	for(int i=1;i<=n;i++)
	{
		cout<<"num: "<<nodelist[i].num<<" weight is: "<<nodelist[i].weight<<" level is: "<<nodelist[i].level<<endl;
	 } 
	*/
	int k=callevel(n);
	//cout<<"k: "<<k<<endl;
	long long int sum[k];
	for(int i=1;i<=k;i++)
		sum[i]=0;
	for(int i=1;i<=n;i++)
	{
		sum[nodelist[i].level]+=nodelist[i].weight;
	}
//	for(int i=1;i<=k;i++)
	//{
	///	cout<<"sum["<<i<<"] is "<<sum[i]<<endl;
//	}
	long long int min=10000000000;
	//cout<<"min: "<<min<<endl;
	int temp=0;
	for(int i=1;i<=k;i++)
	{
		if(sum[i]<=min)
			{
				min=sum[i];
				temp=i;
			}
	}
	cout<<temp;
	return 0;
}

int callevel(int k)
{
	for(int i=1;i<=20;i++)
	{
		if(k<pow(2,i)&&k>=pow(2,i-1))
			return i;
	}
}