原题:
题目描述 给定一棵包含N个节点的完全二叉树,树上每个节点都有一个权值,按从上到下。从左到右的顺序依次是A1,AN,如下图所示:
现在小明要把相同深度的节点的权值加在一起,他想知道哪个深度的节点权值之和最大?如果有多个深度的权值和同为最大,请你输出其中最小的深度。 注:根的深度是1。 输入描述 第一行包含一个整数N(1<N<10^5)。 第二行包含N个整数A1,A…An(-10^5<A≤10^5)。 输出描述 输出一个整数代表答案。
废话不多说,直接上代码:
所有思路都体现在代码的注释里面了。
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 " 的权值和最大"); } }
我正在参与2023腾讯技术创作特训营第三期有奖征文,组队打卡瓜分大奖!