原题:
题目描述 给定一棵包含N个节点的完全二叉树,树上每个节点都有一个权值,按从上到下。从左到右的顺序依次是A1,AN,如下图所示:
现在小明要把相同深度的节点的权值加在一起,他想知道哪个深度的节点权值之和最大?如果有多个深度的权值和同为最大,请你输出其中最小的深度。 注:根的深度是1。 输入描述 第一行包含一个整数N(1<N<10^5)。 第二行包含N个整数A1,A…An(-10^5<A≤10^5)。 输出描述 输出一个整数代表答案。
废话不多说,直接上代码:
所有思路都体现在代码的注释里面了。
代码语言:javascript复制import java.util.Scanner;
public class day04 {
public static void main(String[] args) {
Scanner scanner = new Scanner(System.in);
//输入一个整数N,代表这个完全二叉树有N个节点
int N = scanner.nextInt();
//新建一个长度为N的数组存放节点的数据
int [] N_add = new int[N];
int k = 0;
//flag1代表这个完全二叉树的深度
int flag1 = 1;
while(k<N){
N_add[k ] = scanner.nextInt();
//如果1,3,7,15,也就是到下一深度的节点了,此时要把深度 1;
if(k==Math.pow(2,flag1)-1){
flag1 ;
}
}
//新建一个数组用来存放每一层节点的权值和,长度自然为完全二叉树的深度
int[] i_add = new int[flag1];
flag1 = 0;
for(k = 0;k<N;k ){
//如果1,3,7,15,也就是到下一深度的节点了,此时要把深度 1;
if ((Math.pow(2,flag1) - 1) == k){
flag1 ;
}
//我是先处理了深度,但是存放却是前一层的权值和,因此这里的索引应当减一
i_add[flag1-1] = N_add[k];
}
flag1 = 1;
for(int i:i_add)
System.out.println("深度为: " (flag1 ) " 的权值和是: " i);
int max_i = i_add[0];
int result = 1;
for(int i = 0;i<i_add.length;i ){
if(max_i<i_add[i]) {
//更新最大权值和
max_i = i_add[i];
//因为数组下标是从0开始的,而对应的深度是从1开始的,所以这里的结果应当加1
result = i 1;
}
}
System.out.println("深度为: " result " 的权值和最大");
}
}