三等分数组,使每段数组和相等

import java.util.Scanner;

public class Test {
    public static void main(String[] args) {
        Scanner scanner = new Scanner(System.in);
        String input = scanner.nextLine();
        if (input == null){
            return;
        }
        String[] nums = input.split(" ");
//        int[] nums = {2, 1, 3, 1, 1, 1, 5, 3};
        int[] arr = new int[nums.length];
        for (int i = 0; i < nums.length; i++) {
            arr[i] = Integer.parseInt(nums[i]);
        }
        boolean meth = meth(arr);
        System.out.println(meth);
    }

    public static boolean meth(int[] arr) {
        if (null == arr || arr.length < 4) {
            return false;
        }
        int startIndex = 1;
        int endIndex = arr.length - 2;
        int sum = 0;
        int fSum = arr[0];
        int eSum = arr[arr.length - 1];
        int mSum = 0;
        for (int i = 0; i < arr.length; i++) {
            sum += arr[i];
        }
        mSum = sum - fSum - eSum - arr[startIndex] - arr[endIndex];

        while (fSum < eSum) {
            fSum += arr[startIndex];
            startIndex++;
            if (startIndex > endIndex - 2) {
                return false;
            }
            mSum = sum - fSum - eSum - arr[startIndex] - arr[endIndex];
            continue;
        }
        while (fSum > eSum) {
            eSum += arr[endIndex];
            endIndex--;
            if (startIndex > endIndex - 2) {
                return false;
            }
            mSum = sum - fSum - eSum - arr[startIndex] - arr[endIndex];
            continue;
        }

        while (startIndex <= endIndex - 2) {
            if (fSum == eSum && eSum == mSum) {
                System.out.println("first index : " + startIndex + ", next index : " + endIndex);
                return true;
            }
            eSum += arr[endIndex];
            endIndex--;
            if (startIndex == endIndex - 2) {
                fSum += arr[startIndex];
                startIndex++;
                endIndex = arr.length - 2;
                eSum = arr[arr.length - 1];
                mSum = sum - fSum - eSum - arr[startIndex] - arr[endIndex];
            }
        }
        return false;
    }
}

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

MSK_LY

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值