leetcode548

最新推荐文章于 2024-07-12 18:49:12 发布

爱打篮球的憨憨

最新推荐文章于 2024-07-12 18:49:12 发布

阅读量465

点赞数

CC 4.0 BY-SA版权

分类专栏： leetcode 文章标签： leetcode

本文链接：https://blog.youkuaiyun.com/xiaobailaji/article/details/119725568

leetcode 专栏收录该内容

61 篇文章

订阅专栏

该博客探讨了一种通过三层循环和前缀和优化的算法来解决数组划分的问题。博主首先介绍了暴力求解方法，然后提出使用前缀数组减少循环层数，提高效率。在循环优化中，博主展示了如何调整循环顺序，利用集合存储中间结果，以判断是否存在相等的子数组和。这种方法对于处理大数据集时能显著提升性能。

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

在这里插入图片描述
暴力求解即可，三层遍历循环
优化的方法：
前缀数组：缩短了求和的时间
i、j、k将j作为外层循环，i和k并行的作为内部循环，减少了循环的层数，缩短了时间

public class test {
    public boolean splitArray(int[] nums){
        int len = nums.length;
        int[] prefixSum = new int[len];

        for(int i=0; i<len; i++){
            if(i==0){
                prefixSum[i] = nums[i];
            }else{
                prefixSum[i] += prefixSum[i-1] + nums[i];
            }
        }

        for(int j=3; j<len-3; j++){
            HashSet<Integer> set = new HashSet<>();
            for(int i=1; i<j-1; i++){
                int sum1 = prefixSum[i-1];
                int sum2 = prefixSum[j-1] - prefixSum[i];
                if(sum1 == sum2){
                    set.add(sum1);
                }
            }

            for(int k=j+2; k<len-1; k++){
                int sum3 = prefixSum[k-1] - prefixSum[j];
                int sum4 = prefixSum[len-1] - prefixSum[k];
                if(sum3 == sum4 && set.contains(sum3)){
                    return true;
                }
            }
        }
        return false;
    }
}