河南省藍橋杯第十屆省賽-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;
	}
}