完全二叉树的权值

2022-11-29 08:39:33 浏览数 (1)

原题:

题目描述 给定一棵包含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 " 的权值和最大");
    }
}

0 人点赞