前往小程序,Get更优阅读体验!
立即前往
首页
学习
活动
专区
工具
TVP
发布
社区首页 >专栏 >LeetCode 5981. 分组得分最高的所有下标(前缀和)

LeetCode 5981. 分组得分最高的所有下标(前缀和)

作者头像
Michael阿明
发布2022-03-10 18:33:55
2280
发布2022-03-10 18:33:55
举报

文章目录

1. 题目

给你一个下标从 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 的个数之 和 。

返回 分组得分 最高 的 所有不同下标 。你可以按 任意顺序 返回答案。

代码语言:javascript
复制
示例 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

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

2. 解题

  • 正向前缀和求左侧 0 的个数
  • 反向前缀和求右侧 1 的个数
  • 遍历求最大的和、及其 下标
代码语言:javascript
复制
class Solution:
    def maxScoreIndices(self, nums: List[int]) -> List[int]:
        n = len(nums)
        left0, right1 = [0 for _ in range(n)], [0 for _ in range(n)]
        for i in range(n):
            left0[i] = (0 if i==0 else left0[i-1]) + int(nums[i]==0)
        for i in reversed(range(n)):
            right1[i] = (0 if i==n-1 else right1[i+1]) + int(nums[i]==1)
        maxscore = 0
        idxs = []
        for i in range(n+1):
            score = (0 if i==0 else left0[i-1]) + (0 if i==n else right1[i])
            if score > maxscore:
                maxscore = score
                idxs = [i]
            elif score == maxscore:
                idxs.append(i)
        return idxs

1376 ms 30 MB Python3


我的CSDN博客地址 https://michael.blog.csdn.net/

本文参与 腾讯云自媒体同步曝光计划,分享自作者个人站点/博客。
如有侵权请联系 cloudcommunity@tencent.com 删除

本文分享自 作者个人站点/博客 前往查看

如有侵权,请联系 cloudcommunity@tencent.com 删除。

本文参与 腾讯云自媒体同步曝光计划  ,欢迎热爱写作的你一起参与!

评论
登录后参与评论
0 条评论
热度
最新
推荐阅读
目录
  • 文章目录
  • 1. 题目
  • 2. 解题
领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档