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;
}
}
运行截图
标签:得分,下标,nums,int,08,leetcode.2155,numsright,numsleft,2022.02 来源: https://blog.csdn.net/qq_60717329/article/details/122821955