【拒绝算法PUA】LeetCode 2270. 分割数组的方案数

系列文章目录

【拒绝算法PUA】0x00-位运算
【拒绝算法PUA】0x01- 区间比较技巧
【拒绝算法PUA】0x02- 区间合并技巧
【拒绝算法PUA】0x03 - LeetCode 排序类型刷题
【拒绝算法PUA】LeetCode每日一题系列刷题汇总-2025年持续刷新中

C++刷题技巧总结：
[温习C/C++]0x04 刷题基础编码技巧

---

---

LeetCode 2270. 分割数组的方案数

难度：简单

链接

2270. 分割数组的方案数

题目

给你一个下标从 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 <= 105
-105 <= nums[i] <= 105

解题方法1 - 前缀和

#include <iostream>
#include <vector>
#include <numeric>

using namespace std;

const int N = 1e6;

class Solution {
public:
    int waysToSplitArray(vector<int> &nums) {
        //if empty
        if (nums.size() == 0) {
            return 0;
        }

        //otherwise
        long long presum[N] = {0};
        long long res = 0;
        presum[0] = nums[0];
        //初始化前缀和
        for (int i = 1; i < nums.size(); i++) {
            presum[i] = presum[i - 1] + nums[i];
        }
        // 所有数字求和;
        long long sum = presum[nums.size() - 1];

        //遍历全缀合
        for (int i = 0; i < nums.size() - 1; i++) {
            long long leftSum = presum[i];
            long long rightSum = sum - leftSum;
            if (leftSum >= rightSum) {
                res++;
            }
        }
        return res;
    }
};

int main(int argc, char **argv) {
    vector<int> vec = {10, 4, -8, 7};
    Solution obj;
    int res = obj.waysToSplitArray(vec);
    cout << res << endl;
    return 0;
}

• 输出

3

---

关注我，跟我一起每日一题！
【拒绝算法PUA】LeetCode每日一题系列刷题汇总-2025年持续刷新中