【算法题】2270. 分割数组的方案数

2023-03-12  本文已影响0人  程序员小2

题目:

给你一个下标从 0 开始长度为 n 的整数数组 nums 。
如果以下描述为真,那么 nums 在下标 i 处有一个 合法的分割 :

前 i + 1 个元素的和 大于等于 剩下的 n - i - 1 个元素的和。
下标 i 的右边 至少有一个 元素,也就是说下标 i 满足 0 <= i < n - 1 。
请你返回 nums 中的 合法分割 方案数。

示例 1:

输入:nums = [10,4,-8,7]
输出:2
解释:
总共有 3 种不同的方案可以将 nums 分割成两个非空的部分:

输入:nums = [2,3,1,0]
输出:2
解释:
总共有 2 种 nums 的合法分割:

提示:

2 <= nums.length <= 10^5
-10^5 <= nums[i] <= 10^5

java代码:

class Solution {
  public int waysToSplitArray(int[] nums) {
    int n = nums.length, cnt = 0;
    long[] sum = new long[n + 1];
    for (int i = 0; i < n; i++) {
      sum[i + 1] = sum[i] + nums[i];
    }
    for (int i = 0; i < n - 1; i++) {
      if (sum[i + 1] * 2 >= sum[n]) cnt++;
    }
    return cnt;
  }
}
上一篇 下一篇

猜你喜欢

热点阅读