今天又学会了一个题,想和大家分享一下
题目见下:
给定一棵包含 N 个节点的完全二叉树,树上每个节点都有一个权值,按从 上到下、从左到右的顺序依次是 A1, A2, ··· AN如下图所示:
现在小明要把相同深度的节点的权值加在一起,他想知道哪个深度的节点 权值之和最大?如果有多个深度的权值和同为最大,请你输出其中最小的深度。
注:根的深度是 1。
第一行包含一个整数 N(1N 0^5)N(1≤N≤105)。
第二行包含 NN 个整数 A_1, A_2, ··· A_N (1≤N≤105)
输出一个整数代表答案。
示例
输入
7 1 6 5 4 3 2 12
此题可用队列知识来解决,但是此题很多细节,千万要小心!!!
#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; }