[LeetCode in Python] 5392 (E) maximum score after splitting a string 分割字符串的最大得分

题目

https://leetcode-cn.com/problems/maximum-score-after-splitting-a-string/

给你一个由若干 0 和 1 组成的字符串 s ,请你计算并返回将该字符串分割成两个 非空 子字符串(即 左 子字符串和 右 子字符串)所能获得的最大得分。
「分割字符串的得分」为 左 子字符串中 0 的数量加上 右 子字符串中 1 的数量。

示例 1:

输入:s = "011101"
输出:5

解释:
将字符串 s 划分为两个非空子字符串的可行方案有:
左子字符串 = "0" 且 右子字符串 = "11101",得分 = 1 + 4 = 5
左子字符串 = "01" 且 右子字符串 = "1101",得分 = 1 + 3 = 4
左子字符串 = "011" 且 右子字符串 = "101",得分 = 1 + 2 = 3
左子字符串 = "0111" 且 右子字符串 = "01",得分 = 1 + 1 = 2
左子字符串 = "01110" 且 右子字符串 = "1",得分 = 2 + 1 = 3

示例 2:

输入:s = "00111"
输出:5

解释:当 左子字符串 = "00" 且 右子字符串 = "111" 时,我们得到最大得分 = 2 + 3 = 5

示例 3:

输入:s = "1111"
输出:3

提示:

2 <= s.length <= 500
字符串 s 仅由字符 '0' 和 '1' 组成。

解题思路

  • 记left_sum_list[i]表示s[:i]的得分

  • 记right_sum_list[i]表示s[i:]的得分

  • 为了代码简单粗暴好懂,拆分为3步

  • 第1步,从左向右,更新left_sum_list数组

  • 第2步,从右向左,更新right_sum_list数组

  • 第3步,两两求和,找最大值

  • 没有使用zip()等函数,是为了更好懂

  • left_sum_list和right_sum_list初始化为n+1的数组,也是为了避免边界判断

代码

class Solution:
    def maxScore(self, s: str) -> int:
        n = len(s)
        left_sum_list = [0] * (n+1)
        right_sum_list = [0] * (n+1)

        # - update left sum list
        for i in range(1,n):
            if s[i-1] == '0': 
                left_sum_list[i] = left_sum_list[i-1] + 1
            else:
                left_sum_list[i] = left_sum_list[i-1]
        
        # - update right sum list
        for i in range(n-1,0,-1):
            if s[i] == '1': 
                right_sum_list[i] = right_sum_list[i+1] + 1
            else:
                right_sum_list[i] = right_sum_list[i+1]

        # - get max sum
        max_sum = 0
        for i in range(1,n):
            max_sum = max(max_sum, left_sum_list[i] + right_sum_list[i])

        return max_sum
posted @ 2020-04-26 16:38  ET民工[源自火星]  阅读(124)  评论(0编辑  收藏