1422. 分割字符串的最大得分 :「前缀和」&「模拟」

题目描述

这是 LeetCode 上的 1422. 分割字符串的最大得分 ,难度为 简单

Tag : 「前缀和」、「模拟」

给你一个由若干 01 组成的字符串 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' 组成。

前缀和

构建前缀和数组来记录每个前缀中 $1$ 个个数,复杂度为 $O(n)$,枚举每个分割点,搭配前缀和数组计算左串中 $0$ 的数量和右串中 $1$ 的数量,取所有得分的最大值即是答案。

Java 代码:

class Solution {
    public int maxScore(String s) {
        int n = s.length(), ans = 0;
        int[] sum = new int[n + 10];
        for (int i = 1; i <= n; i++) sum[i] = sum[i - 1] + (s.charAt(i - 1) - '0');
        for (int i = 1; i <= n - 1; i++) {
            int a = i - sum[i], b = sum[n] - sum[i];
            ans = Math.max(ans, a + b);
        }
        return ans;
    }
}

Typescript 代码:

function maxScore(s: string): number {
    let n = s.length, ans = 0
    const sum = new Array(n + 10).fill(0)
    for (let i = 1; i <= n; i++) sum[i] = sum[i - 1] + (s[i - 1] == '1' ? 1 : 0)
    for (let i = 1; i <= n - 1; i++) {
        const a = i - sum[i], b = sum[n] - sum[i]
        ans = Math.max(ans, a + b)
    }
    return ans
};
  • 时间复杂度:$O(n)$
  • 空间复杂度:$O(n)$

模拟

更进一步,利用 s 中有只有 $0$ 和 $1$,我们可以遍遍历 s 边计算得分(而无须预处理前缀和数组),起始分割点为 s[0],此时得分为 s[0] 中 $0$ 的个数加上 s[1...(n-1)] 中 $1$ 的个数。

然后继续往后处理 s,当 $s[i] = 0$,说明有一个 $0$ 从右串中移到了左串,并且 $0$ 在右串中不得分,在左串中得分,因此总得分加一;而当 $s[i] = 1$,说明有一个 $1$ 从右串中移到了左串,而 $1$ 在右串中得分,在左串中不得分,因此总得分减一。在所有得分中取最大值即是答案。

Java 代码:

class Solution {
    public int maxScore(String s) {
        int n = s.length(), cur = s.charAt(0) == '0' ? 1 : 0;
        for (int i = 1; i < n; i++) cur += s.charAt(i) - '0';
        int ans = cur;
        for (int i = 1; i < n - 1; i++) {
            cur += s.charAt(i) == '0' ? 1 : -1;
            ans = Math.max(ans, cur);
        }
        return ans;
    }
}

TypeScript 代码:

function maxScore(s: string): number {
    let n = s.length, cur = s[0] == '0' ? 1 : 0
    for (let i = 1; i < n; i++) cur += s[i] == '1' ? 1 : 0
    let ans = cur
    for (let i = 2; i < n - 1; i++) {
        cur += s[i] == '0' ? 1 : -1
        ans = Math.max(ans, cur)
    }
    return ans
};
  • 时间复杂度:$O(n)$
  • 空间复杂度:$O(1)$

最后

这是我们「刷穿 LeetCode」系列文章的第 No.1422 篇,系列开始于 2021/01/01,截止于起始日 LeetCode 上共有 1916 道题目,部分是有锁题,我们将先把所有不带锁的题目刷完。

在这个系列文章里面,除了讲解解题思路以外,还会尽可能给出最为简洁的代码。如果涉及通解还会相应的代码模板。

为了方便各位同学能够电脑上进行调试和提交代码,我建立了相关的仓库:https://github.com/SharingSou...

在仓库地址里,你可以看到系列文章的题解链接、系列文章的相应代码、LeetCode 原题链接和其他优选题解。

更多更全更热门的「笔试/面试」相关资料可访问排版精美的 合集新基地

本文由mdnice多平台发布

你可能感兴趣的:(后端)