携手创作，共同成长！这是我参与「掘金日新计划 · 8 月更文挑战」的第14天，点击查看活动详情 >>

分组得分最高的所有下标

给你一个下标从 0 开始的二进制数组 nums ，数组长度为 n 。nums 可以按下标 i（ 0 <= i <= n ）拆分成两个数组（可能为空）：numsleft 和 numsright 。

numsleft 包含 nums 中从下标 0 到 i - 1 的所有元素（包括 0 和 i - 1 ），而 numsright 包含 nums 中从下标 i 到 n - 1 的所有元素（包括 i 和 n - 1 ）。如果 i == 0 ，numsleft 为空，而 numsright 将包含 nums 中的所有元素。如果 i == n ，numsleft 将包含 nums 中的所有元素，而 numsright 为空。下标 i 的分组得分为 numsleft 中 0 的个数和 numsright 中 1 的个数之和。

返回分组得分最高的所有不同下标。你可以按任意顺序返回答案。

示例1:

输入：nums = [0,0,1,0]
输出：[2,4]
解释：按下标分组
- 0 ：numsleft 为 [] 。numsright 为 [0,0,1,0] 。得分为 0 + 1 = 1 。
- 1 ：numsleft 为 [0] 。numsright 为 [0,1,0] 。得分为 1 + 1 = 2 。
- 2 ：numsleft 为 [0,0] 。numsright 为 [1,0] 。得分为 2 + 1 = 3 。
- 3 ：numsleft 为 [0,0,1] 。numsright 为 [0] 。得分为 2 + 0 = 2 。
- 4 ：numsleft 为 [0,0,1,0] 。numsright 为 [] 。得分为 3 + 0 = 3 。
下标 2 和 4 都可以得到最高的分组得分 3 。
注意，答案 [4,2] 也被视为正确答案。

示例2:

输入：nums = [0,0,0]
输出：[3]
解释：按下标分组
- 0 ：numsleft 为 [] 。numsright 为 [0,0,0] 。得分为 0 + 0 = 0 。
- 1 ：numsleft 为 [0] 。numsright 为 [0,0] 。得分为 1 + 0 = 1 。
- 2 ：numsleft 为 [0,0] 。numsright 为 [0] 。得分为 2 + 0 = 2 。
- 3 ：numsleft 为 [0,0,0] 。numsright 为 [] 。得分为 3 + 0 = 3 。
只有下标 3 可以得到最高的分组得分 3 。

实例3:

输入：nums = [1,1]
输出：[0]
解释：按下标分组
- 0 ：numsleft 为 [] 。numsright 为 [1,1] 。得分为 0 + 2 = 2 。
- 1 ：numsleft 为 [1] 。numsright 为 [1] 。得分为 0 + 1 = 1 。
- 2 ：numsleft 为 [1,1] 。numsright 为 [] 。得分为 0 + 0 = 0 。
只有下标 0 可以得到最高的分组得分 2 。

提示:

n == nums.length
1 <= n <= 105
nums[i] 为 0 或 1

解题思路:

把i看做一个指针，从左向右移动的过程中如果指针指向0 则在左数组的0数量++ 指向1则右数组中的1数量--

我的答案:

/**
 * @param {number[]} nums
 * @return {number[]}
 */
var maxScoreIndices = function(nums) {
    var numFor1 = 0; // 右数组中所有1的数量
    var numFor0 = 0; // 左数组中0的数量
    for(var i=0;i<nums.length;i++) {
        if(nums[i] === 1) {
            numFor1++
        }
    }
    var maxNum = numFor1;  // 记录最大值;
    var res = [0]; // 维护一个结果数组，在maxNum新的pos值大于等于的时候更新
    for(var i=1;i<=nums.length;i++) {
        if(nums[i-1] === 0) {
            numFor0++;
        } else {
            numFor1--;
        }
        // 注意大于和等于是两种处理
        if(numFor0 + numFor1 > maxNum) {
            maxNum = numFor0 + numFor1;
            res = [i];
        } else if (numFor0 + numFor1 === maxNum) {
            res.push(i);
        }        
    }
    return res;
};

最后

如果有更好的解法或者思路, 欢迎在评论区和我交流~ ღ( ´･ᴗ･` )

LeetCode Everyday - 分组得分最高的所有下标