LeetCode 2155. 分组得分最高的所有下标

一、题目

1、题目描述

  给你一个下标从 $0$ 开始的二进制数组 nums，数组长度为 $n$ 。nums可以按下标 $i（0\le i\le n）$ 拆分成两个数组（可能为空）：numsleft和 numsright。numsleft包含 nums中从下标 $0$ 到 $i-1$ 的所有元素（包括 $0$ 和 $i-1$ ），而 numsright包含 nums中从下标 $i$ 到 $n-1$ 的所有元素（包括 $i$ 和 $n-1$ ）。如果 i == 0，numsleft 为 空。如果 i == n，而 numsright 为 空 。下标 $i$ 的 分组得分 为 numsleft中 $0$ 的个数和 numsright中 $1$ 的个数之 和 。返回 分组得分 最高 的 所有不同下标 。你可以按 任意顺序 返回答案。
  样例输入： nums = [0,0,1,0]
  样例输出： [2,4]

2、基础框架

• C语言 版本给出的基础框架代码如下：

int* maxScoreIndices(int* nums, int numsSize, int* returnSize){
}

3、原题链接

LeetCode 2155. 分组得分最高的所有下标

二、解题报告

1、思路分析

  $(1)$ 首先，需要建立两个数组：sum0[i]代表原数组前缀零的个数；sum1[i]代表原数组后缀一的个数。
  $(2)$顺序统计前缀零；逆序统计后缀一。
  $(3)$ 对于所有的下标 $i(0\le i\le n)$，求 $sum0[i-1]+sum1[i]$ 的最大值，然后保留下标 $i$，并且将所有的下标 $i$ 组成一个列表返回。
  $(4)$ 注意边界条件，$sum0[-1]=0$，考虑数组下标越界。

2、时间复杂度

  $O(n)$。

3、代码详解

/**
 * Note: The returned array must be malloced, assume caller calls free().
 */

int sum[2][100010];

int getSum(int idx, int i) {
    if(i == -1) {
        return 0;
    }
    return sum[idx][i];
}

int* maxScoreIndices(int* nums, int n, int* returnSize){
    int i;
    int *ret = (int *)malloc( sizeof(int) * n );
    int tmp, max;
    *returnSize = 0;
    for(i = 0; i < n; ++i) {
        sum[0][i] = (nums[i] == 0) ? 1 : 0;
        if(i) {
            sum[0][i] += sum[0][i-1];
        }
    }
    sum[1][n] = 0;
    for(i = n - 1; i >= 0; --i) {
        sum[1][i] = (nums[i] == 1) ? 1 : 0;
        sum[1][i] += sum[1][i+1];
    }

    max = -1;
    for(i = 0; i <= n; ++i) {
        tmp = getSum(0, i-1) + getSum(1, i);
        if(tmp > max) {
            max = tmp;
            ret[0] = i;
            (*returnSize) = 1;
        }else if(tmp == max) {
            ret[ (*returnSize)++ ] = i;
        }
    }
    return ret;
}

---

三、本题小知识

  前缀和 + 后缀和。

---

四、加群须知

  相信看我文章的大多数都是「 大学生 」，能上大学的都是「 精英 」，那么我们自然要「 精益求精 」，如果你还是「 大一 」，那么太好了，你拥有大把时间，当然你可以选择「 刷剧 」，然而，「 学好算法 」，三年后的你自然「 不能同日而语 」。
  那么这里，我整理了「 几十个基础算法 」 的分类，点击开启：

🌌《算法入门指引》🌌
  如果链接被屏蔽，或者有权限问题，可以私聊作者解决。

  大致题集一览：

  为了让这件事情变得有趣，以及「 照顾初学者 」，目前题目只开放最简单的算法 「 枚举系列 」 （包括：线性枚举、双指针、前缀和、二分枚举、三分枚举），当有 一半成员刷完 「 枚举系列 」 的所有题以后，会开放下个章节，等这套题全部刷完，你还在群里，那么你就会成为「 夜深人静写算法 」专家团 的一员。
  不要小看这个专家团，三年之后，你将会是别人 望尘莫及 的存在。如果要加入，可以联系我，考虑到大家都是学生， 没有「 主要经济来源 」，在你成为神的路上，「 不会索取任何 」。
  🔥联系作者，或者扫作者主页二维码加群，加入刷题行列吧🔥

---

🔥让天下没有难学的算法🔥
C语言免费动漫教程，和我一起打卡！ 🌞《光天化日学C语言》🌞
让你养成九天持续刷题的习惯 🔥《九日集训》🔥
入门级C语言真题汇总 🧡《C语言入门100例》🧡
组团学习，抱团生长 🌌《算法零基础100讲》🌌
几张动图学会一种数据结构 🌳《画解数据结构》🌳
竞赛选手金典图文教程 💜《夜深人静写算法》💜