2022.02.08_每日一题 leetcode.2155

题目描述

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

给你一个下标从 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 <= 10^5
nums[i] 为 0 或 1

思路

先遍历一遍数组 ，模拟指针在 0 时候的情况 ，此时右边数组的和默认设为初始得分最高的最大值 。之后按照for 循环模拟指针移动 ，依据 nums[i]的值来确定移动时 leftCnt 和 rightCnt 的变化 ，max 记录最高得分 ，如果得分高于最高得分则max更新 ，集合清空，重新存储下标，如果相等则直接添加

代码

class Solution {
    public List<Integer> maxScoreIndices(int[] nums) {
        List<Integer> list = new ArrayList<>();
        int leftCnt = 0; // 左边数组中 0 的数量
        int rightCnt = 0; // 右边数组中 1 的数量
        for (int num : nums) {
            rightCnt += num;
        }
        int len = nums.length;
        int max = rightCnt;
        list.add(0);
        for (int i = 0; i < len; i++) {
            if (nums[i] == 0) {
                leftCnt++;
            }else {
                rightCnt--;
            }
            int sum = leftCnt + rightCnt;
            if (sum > max){
                max = sum;
                list.clear();
                list.add(i + 1);
            }else if (sum == max){
                list.add(i + 1);
            }
        }
        return list;
    }
}

运行截图