1422. 分割字符串的最大得分

Problem#

给你一个由若干 0 和 1 组成的字符串 s，请你计算并返回将该字符串分割成两个非空子字符串（即左子字符串和右子字符串）所能获得的最大得分。

「分割字符串的得分」为：左子字符串中 0 的数量 + 右子字符串中 1 的数量。

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

示例 1：

输入：s = “011101” 输出：5 解释：所有可能的分割为：左 = “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

约束：

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

思路#

先统计总 1 的数量，然后遍历分割点，左边 0 递增，右边 1 递减，取最大值。

1
class Solution:
2
    def maxScore(self, s: str) -> int:
3
        total_ones = s.count('1')
4
        zeros = 0
5
        ones = total_ones
6
        max_score = 0
7
        for i in range(len(s) - 1):
8
            if s[i] == '0':
9
                zeros += 1
10
            else:
11
                ones -= 1
12
            max_score = max(max_score, zeros + ones)
13
        return max_score

1
import unittest
2

3

4
class TestMaxScore(unittest.TestCase):
5
    def setUp(self):
6
        self.solution = Solution()
7

8
    def test_example1(self):
9
        self.assertEqual(self.solution.maxScore("011101"), 5)
10

11
    def test_example2(self):
12
        self.assertEqual(self.solution.maxScore("00111"), 5)
13

14
    def test_all_zeros(self):
15
        self.assertEqual(self.solution.maxScore("00"), 1)
16

17
    def test_all_ones(self):
18
        self.assertEqual(self.solution.maxScore("11"), 1)
19

20

21
if __name__ == '__main__':
22
    unittest.main()