历届真题完全二叉树的权值【第十届】【省赛】【A组】 java

原创于 2022-03-16 15:15:21 发布 · 977 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#java #贪心算法 #算法

这篇博客主要介绍了一个利用二叉树和前缀和解决区间最大值问题的算法。代码中，作者通过计算前缀和数组，并结合二分查找的思想，实现了在O(logn)的时间复杂度内找到数组中某个区间的最大值。算法适用于处理大规模数据，具有较高的效率。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

没仔细学过二叉树，写麻烦了。。

AC代码


import java.util.Scanner;

public class Main {
    public static void main(String[] args) {
        Scanner sc = new Scanner(System.in);
        int N = sc.nextInt();
        int[] nums = new int[N+1];
        int[] sum = new int[N+1];
        //前缀和
        for(int i = 1;i<=N;i++){
            nums[i] = sc.nextInt();
            sum[i] = sum[i-1] + nums[i];
        }

        double pos = N+1;
        int ks = (int) (Math.log(pos)/Math.log(2));
        int id = (int) (Math.pow(2,ks) - 1);
        int later = sum[N] - sum[id];
        int max = later;
        int ans = ks+1;
        while(id > 1){
            double l = id + 1 ;
            int k = (int) (Math.log(l)/Math.log(2));//深度
            int x = (int) Math.pow(2,k-1);
            int j = id - x;
            if(sum[id]-sum[j] >= max){
                max = sum[id]-sum[j];
                ans = k;
            }
            id = j;
        }
        if(nums[1] >= max){
            ans = 1;
        }
        System.out.println(ans);
    }
}