持续创作，加速成长！这是我参与「掘金日新计划 · 6 月更文挑战」的第24天，点击查看活动详情
1、前言
每天一个算法小练习，本篇使用Java实现。
2、题目描述
给你一个下标从 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 的个数之和 。
返回分组得分最高的所有不同下标。可以按任意顺序返回答案。
2.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.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、解题思路
先遍历一遍数组，得到有多少个1，即numsright的最大得分。然后再次遍历数组，如果nums[i]为0，则numsleft加分，否则numsright减分，将左右分值相加与之前计算的numsright的最大得分比较大小，放入hashmap中，最后从map中取出符合条件的下标数组即可。
3.1、实现代码
public List<Integer> maxScoreIndices(int[] nums) {
//用来存储最大分值及其对于的下标list
HashMap<Integer, List<Integer>> map = new HashMap<>();
//记录最大分值
int max = 0;
//右侧得分
int rightScore = 0;
//左侧得分
int leftScore = 0;
for (int i = 0; i < nums.length; i++) {
if (nums[i] == 1)
rightScore++;
}
max = rightScore;
ArrayList<Integer> maxScoreList = new ArrayList<>();
maxScoreList.add(0);
map.put(rightScore, maxScoreList);
for (int i = 0; i < nums.length; i++) {
if (nums[i] == 0) {
leftScore++;
} else {
rightScore--;
}
int temp = leftScore + rightScore;
max = Math.max(temp, max);
//最大分值放入map中
List<Integer> list = map.getOrDefault(temp, new ArrayList<>());
list.add(i + 1);
map.put(temp, list);
}
return map.get(max);
}
3.2、执行结果
好了、本期就先介绍到这里，有什么需要交流的，大家可以随时私信我。😊
