蓝桥: 完全二叉树的权值 (20分)

蓝桥杯: 完全二叉树的权值 (20分)

给定一棵包含 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

数据范围:

对于所有评测用例,1 ≤ N ≤ 100000,−100000 ≤ A i ≤ 100000。

#include<iostream>
using namespace std;
int main(void){
	int n , cnt=1 ,ans = -1 ,ok=0 ,ansDeep=-1;
	cin>>n; 
	for( int  i=1;i<=n;i++){
		 if(  i==1 )
		      cnt=1;
		 else cnt*=2; 
		 int  sum = 0 ,t;
		 for( int j=1;j<=cnt;j++){
		 	  if( ok >= n )
		 	      break;
			  cin>>t;
		 	  sum+=t;
		 	  ok++; 
		 } 
		 if( ans < sum ){
		 	 ans = sum;
			 ansDeep = i ; 
		 }
		 if( ok == n )
		     break;
	}
	cout<<ansDeep<<endl;
	return 0;
}
发布了742 篇原创文章 · 获赞 132 · 访问量 8万+

猜你喜欢

转载自blog.csdn.net/S_999999/article/details/103967690
今日推荐