藍橋杯：完全二叉樹的權值——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;	
}

運行結果：