题目描述
给你一个由若干 0 和 1 组成的字符串 s
,请你计算并返回将该字符串分割成两个 非空 子字符串(即 左 子字符串和 右 子字符串)所能获得的最大得分。
「分割字符串的得分」为 左 子字符串中 0 的数量加上 右 子字符串中 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
输入:s = "00111"
输出:5
解释:当 左子字符串 = "00" 且 右子字符串 = "111" 时,我们得到最大得分 = 2 + 3 = 5
输入:s = "1111"
输出:3
提示:
2 <= s.length <= 500
- 字符串
s
仅由字符'0'
和'1'
组成。
算法分析
- 找到
'0'
的前缀和数组a[]
,'1'
的后缀和数组b[]
,再枚举每个位置,求出a[i] + b[i + 1]
的最大值即可
时间复杂度 $O(n)$
Java 代码
class Solution {
public int maxScore(String s) {
int n = s.length();
char[] g = new char[n + 2];
for(int i = 1;i <= n;i ++) g[i] = s.charAt(i - 1);
int[] a = new int[n + 2];
int[] b = new int[n + 2];
for(int i = 1;i <= n;i ++)
{
a[i] = a[i - 1];
if(g[i] == '0') a[i] ++;
}
for(int i = n;i >= 1;i --)
{
b[i] = b[i + 1];
if(g[i] == '1') b[i] ++;
}
int res = 0;
for(int i = 1;i <= n - 1;i ++)
{
res = Math.max(res,a[i] + b[i + 1]);
}
return res;
}
}