[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