将数组分为n个相等的部分

给定一个正数数组arr,返回该数组能不能分成4个部分,并且每个部分的累加和相等,切分位置的数不要。 例如: arr=[3, 2, 4, 1, 4, 9, 5, 10, 1, 2, 2] 返回true 三个切割点下标为2, 5, 7. 切出的四个子数组为[3,2], [1,4], [5], [1,2,2], 累加和都是5 。

import java.util.HashMap;
import java.util.Map;

public class Main {

    private static boolean isOk(int[] arr, int k) {
        if (arr == null || arr.length < 7) {
            return false;
        }
        int n = arr.length;

        int[] sum = new int[n];
        Map<Integer, Integer> hash = new HashMap<>();
        hash.put(0, -1);
        for (int i = 0; i < n; ++i) {
            sum[i] = i == 0 ? arr[i] : sum[i - 1] + arr[i];
            hash.put(sum[i], i);
        }

        boolean ok = false;

        // 第一个切分枚举的最后位置
        int last = n - (k - 1) * 2;

        // 0 1 2 3 4 5 6 7
        // 枚举第一个切分
        for (int i = 1; i <= last; ++i) {
            // 一部分的累加和
            int net = sum[i - 1];
            // 当前有1个部分,最终目标为k个累加和为k的部分
            int count = 1;
            // 当前切分点
            int index = i;

            while (count != k && index < n - 1) {
                // 获取下一个累加和为net的截止位置
                Integer netIndex = hash.get(net + sum[index]);
                // 不存在则尝试失败
                if (netIndex == null) {
                    break;
                }
                // 更新当前切分点
                index = netIndex + 1;
                ++count;
            }

            // 达到目标数且数组切分完毕
            if (count == k && index == n) {
                ok = true;
                break;
            }

        }
        return ok;
    }

    public static void main(String[] args) {
        int[] arr = {3, 2, 4, 1, 4, 9, 5, 10, 1, 2, 2};
        System.out.println(isOk(arr, 4));
    }
}
posted @   Tianyiya  阅读(503)  评论(0编辑  收藏  举报
编辑推荐:
· 开发者必知的日志记录最佳实践
· SQL Server 2025 AI相关能力初探
· Linux系列:如何用 C#调用 C方法造成内存泄露
· AI与.NET技术实操系列(二):开始使用ML.NET
· 记一次.NET内存居高不下排查解决与启示
阅读排行:
· 被坑几百块钱后,我竟然真的恢复了删除的微信聊天记录!
· 没有Manus邀请码?试试免邀请码的MGX或者开源的OpenManus吧
· 【自荐】一款简洁、开源的在线白板工具 Drawnix
· 园子的第一款AI主题卫衣上架——"HELLO! HOW CAN I ASSIST YOU TODAY
· Docker 太简单,K8s 太复杂?w7panel 让容器管理更轻松!
点击右上角即可分享
微信分享提示