蓝桥杯：完全二叉树的权值——C++详解

题目描述：

给定一棵包含 N 个节点的完全二叉树，树上每个节点都有一个权值，按从 上到下、从左到右的顺序依次是 A1, A2, ··· AN，如下图所示：
 

现在小明要把相同深度的节点的权值加在一起，他想知道哪个深度的节点 权值之和最大？如果有多个深度的权值和同为最大，请你输出其中最小的深度。 

注：根的深度是 1。

输入：

第一行包含一个整数 N。 
第二行包含 N 个整数 A1, A2, ··· AN 。

输出：

输出一个整数代表答案

代码如下：

#include<iostream>
using namespace std;
int main()
{
	int N;
	cin>>N;
	int A[N+1];		            //下标从1开始 
	for(int i=1;i<=N;++i)
		cin>>A[i];
	int depth=0,deep=1,maxn=0,sum;
	while(N>=(1<<(deep)-1))		    //当该深度节点完整时 
	{
		sum=0;
		for(int i=(1<<(deep-1));i<=(1<<deep)-1;i++)
			sum+=A[i];
		if(sum>maxn)
		{
			maxn=sum;
			depth=deep;
		}
		++deep;
	}
	sum=0;
	for(int i=(1<<(deep-1));i<=N;++i)    //最后一层节点不完整时 
		sum+=A[i];
	if(sum>maxn)
	{
		maxn=sum;
		depth=deep;
	}
	cout<<depth;	
	return 0;	
}

运行结果：