【LeetCode每日一题】2270.分割数组的方案数

凭栏听雨客2023-12-12 12:59

https://leetcode.cn/problems/number-of-ways-to-split-array/description/

题目：

给定一个数组，从下标为 index 的地方切开，左边的数大于右边，保证右边至少有一个数。

思路一：

遍历数组，用prefixArr 记录每个元素的前缀和。
再遍历 prefixArr。
- index 最大只能取到 prefix.length -2。因为要保证右边至少有一个数。
- 左边的数（prefixArr） ≥ 右边的数（prefixArr[prefixArr.length-1]-preArr[i]）

js 复制代码

/**
 * @param {number[]} nums
 * @return {number}
 */
var waysToSplitArray = function(nums) {
    let preFixArr = [];
    let sum = 0
    for(let num of  nums){
        sum+=num;
        preFixArr.push(sum);
    }
    let res = 0;
    for(let i = 0;i<nums.length-1;i++){
        if(preFixArr[i]>=sum-preFixArr[i]){
            res ++;
        }
    }
    return res;

};

思路二：

根据上面的思路可以优化，只记录最大值，没有必要用数组记录每个元素的前缀和，只要每次遍历的时候left++，就能得到每次遍历左边的值。

js 复制代码

/**
 * @param {number[]} nums
 * @return {number}
 */
var waysToSplitArray = function(nums) {
    let sum = 0
    for(let num of  nums){
        sum+=num;
    }
    let res = 0;
    left = 0;
    for(let i = 0;i<nums.length-1;i++){
        left+=nums[i];
        if(left>=sum-left){
            res++;
        }
    }
    return res;

};