蓝桥杯--完全二叉树的权值

今天又学会了一个题,想和大家分享一下

题目见下:

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

蓝桥杯--完全二叉树的权值

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

注:根的深度是 1。

输入描述

第一行包含一个整数 N(1蓝桥杯--完全二叉树的权值N 蓝桥杯--完全二叉树的权值0^5)N(1≤N≤105)。

第二行包含 NN 个整数 A_1, A_2, ··· A_N (1≤N≤105)

输出描述

输出一个整数代表答案。

输入输出样例

示例

输入

7
1 6 5 4 3 2 1
2

此题可用队列知识来解决,但是此题很多细节,千万要小心!!!

#include <iostream>
#include <cstdio>
#include <queue>
using namespace std;
typedef long long ll;
const ll num=-1e10; 
//有负数这要max的初始化的值要小心啊!!!
且小心int会爆!!! 
queue<int>que;
int main(){
	int n,x;
	cin>>n;
	for(int i=0;i<n;i++){
		cin>>x;
		que.push(x);
	}
	//一级一级的区间有[2^(n-1),2^n-1]
	int maxi=0,cnt=1;
	ll sum=0,max=num,b=2;
	for(int i=1;que.size();i++){
		sum+=que.front();
		que.pop() ;
		if(i==b-1){//注意最后一层,不一定是满的 
			if(sum>max){
			max=sum;
			maxi=cnt;
			}
			cnt++;
			b*=2;
			sum=0;
		}
	} 
	if(sum>max){
		maxi=cnt;
	}
	cout<<maxi<<endl;
	return 0;
} 

上一篇:字符串碎片


下一篇:LeetCode-93. 复原 IP 地址