侧边栏壁纸
博主头像
GabrielxD

列車は必ず次の駅へ。では舞台は?私たちは?

  • 累计撰写 471 篇文章
  • 累计创建 108 个标签
  • 累计收到 9 条评论

目 录CONTENT

文章目录

【模拟】分割数组的方案数

GabrielxD
2022-05-15 / 0 评论 / 0 点赞 / 49 阅读 / 601 字 / 正在检测是否收录...

题目

6067. 分割数组的方案数


给你一个下标从 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 分割成两个非空的部分:
- 在下标 0 处分割 nums 。那么第一部分为 [10] ,和为 10 。第二部分为 [4,-8,7] ,和为 3 。因为 10 >= 3 ,所以 i = 0 是一个合法的分割。
- 在下标 1 处分割 nums 。那么第一部分为 [10,4] ,和为 14 。第二部分为 [-8,7] ,和为 -1 。因为 14 >= -1 ,所以 i = 1 是一个合法的分割。
- 在下标 2 处分割 nums 。那么第一部分为 [10,4,-8] ,和为 6 。第二部分为 [7] ,和为 7 。因为 6 < 7 ,所以 i = 2 不是一个合法的分割。
所以 nums 中总共合法分割方案受为 2 。

示例 2:

输入:nums = [2,3,1,0]
输出:2
解释:
总共有 2 种 nums 的合法分割:
- 在下标 1 处分割 nums 。那么第一部分为 [2,3] ,和为 5 。第二部分为 [1,0] ,和为 1 。因为 5 >= 1 ,所以 i = 1 是一个合法的分割。
- 在下标 2 处分割 nums 。那么第一部分为 [2,3,1] ,和为 6 。第二部分为 [0] ,和为 0 。因为 6 >= 0 ,所以 i = 2 是一个合法的分割。

提示:

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

解题

方法一:模拟

思路

遍历数组,统计前面部分的和,大于等于总和的一半即为一个合法分割,计数自增。

代码

class Solution {
    public int waysToSplitArray(int[] nums) {
        int len = nums.length;
        long sum = 0, leftSum = 0;
        for (int num : nums) sum += num;
        int ans = 0;
        for (int i = 0; i < len - 1; i++) {
            leftSum += nums[i];
            if (leftSum * 2 >= sum) ans++;
        }
        return ans;
    }
}
0

评论区