【C语言刷LeetCode】2155. 分组得分最高的所有下标(M)

【

给你一个下标从 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

来源：力扣（LeetCode）
链接：https://leetcode.cn/problems/all-divisions-with-the-highest-score-of-a-binary-array
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

】

来自前缀和思想，或者说分治思想，一个从左计算0的个数，一个从右计算1的个数，然后再计算sum[i]的个数i。

int* maxScoreIndices(int* nums, int numsSize, int* returnSize){
    int preleft[numsSize + 1];
    int prer[numsSize + 1];
    int sum[numsSize + 1];
    int i;
    int tmp = 0;
    int max = 0;
    int *retarr;
    int ridx = 0;

    retarr = (int *)malloc(sizeof(int) * numsSize);
    memset(retarr, 0, sizeof(int) * numsSize);

    preleft[0] = 0;
    // preleft[i + 1] 表示到索引i，包括它共有多少个0;那么0到i-1 就是 preleft[i];
    for (i = 0; i < numsSize; i++) {
        tmp = nums[i] == 0 ? 1 : 0;
        preleft[i + 1] = preleft[i] + tmp; // 0 的个数
    }

    prer[numsSize] = 0;
    // prer[i] 表示从索引i到最后，包括索引i共有多少个1
    for (i = numsSize - 1; i >= 0; i--) {
        prer[i] = prer[i + 1] + nums[i];
    }

    for (i = 0; i <= numsSize; i++) { // 从0 到 numsSize
        sum[i] = preleft[i] + prer[i];
        max = fmax(sum[i], max);
    }

    for (i = 0; i <= numsSize; i++) {
        if (sum[i] == max) {
            retarr[ridx++] = i;
        }
    }

    *returnSize = ridx;
    return retarr;
}

/*
思路就是前缀和
一个从左计算0的个数，一个从右计算1的个数，然后再计算sum[0]的个数
*/