五、题目:完全二叉树权值
给定一棵包含 N 个节点的完全二叉树,树上每个节点都有一个权值,按从 上到下、从左到右的顺序依次是 A1, A2, · · · An
现在小明要把相同深度的节点的权值加在一起,他想知道哪个深度的节点 权值之和最大?如果有多个深度的权值和同为最大,请你输出其中最小的深度。
注:根的深度是 1。
输入:
第一行包含一个整数 N。 第二行包含N个整数A1,A2,··· An。
输出格式:
输出一个整数代表答案。
样例输入:
7
1 6 5 4 3 2 1
样例输出:
2
数据范围:
对于所有评测用例,1 ≤ N≤ 100000,−100000 ≤ Ai≤ 100000。
- 分析思路:
完全二叉树,它的每一层都是满的,每一个节点都有左右子节点。
第一层1个节点,第二次2个节点,第三次2^2个节点,依次类推
题目是要把每一层的节点的权值相加的和进行比较,输出最大的权值和所在的那一层,要是权值出现相同,输出权值相加层数最少的那一层。
- 代码:
package wanquantree;
import java.util.Scanner;
public class Erchatree {
public static void main(String[] args) {
Scanner scanner=new Scanner(System.in);
int n=100000;
int num[]=new int[100005]; //数组存放节点数
long ans[]=new long [100005]; //记录每一层的权值
int height=1; //深度,第一层
int count=1; //count每一层节点数,从第一层开始
int x=1; //每层的节点数,每层都不一样
for(int i=1;i<n;i++) {
num[i]=100000;
}
for(int i=1;i<n;i++) { //i:节点数
if(i>count) {
height++; //层数,深度加1
x*=x; //每层的节点数
count+=x; //总的节点数
}
ans[height]+=num[i]; //每层深度对应的权值和
}
for(int i=1;i<height;i++) {
System.out.println(ans[i]);
}
}
}