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

259 阅读2分钟

题目

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

难度简单

给你一个由若干 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

思路

使用两个变量进行统计,rightOne 表示分割后右边字符串中 1 的个数,leftZero 表示分割后左边字符串中

第一次遍历统计字符串中 1 的个数 -> rightOne

第二次遍历进行字符串的分割。遍历时,遇到 0 则将 leftZero 加 1 表示这个字符此时被分到了左边;同理,遇到 1 则将 rightOne 减 1。遍历过程中 rightOne + leftZero 的最大值就是希望得到的最大得分。

因为已经遍历了所有可能的分割方法,所以算法的正确性是有保障的。

实现

func max(a, b int) int {
    if a > b {
        return a
    }
    return b
}

func maxScore(s string) int {
    maxn := 0
    rightOne := 0
    leftZero := 0

    for _, v := range s {
        if v == '1' {
            rightOne++
        }
    }

    for i := 0 ; i < len(s)-1 ; i++ {
        if s[i] == '0' {
            leftZero++
        } else {
            rightOne--
        }
        maxn = max(maxn, leftZero + rightOne)
    }

    return maxn
}