小扣出去秋游,途中收集了一些红叶和黄叶,他利用这些叶子初步整理了一份秋叶收藏集 leaves, 字符串 leaves 仅包含小写字符 r 和 y, 其中字符 r 表示一片红叶,字符 y 表示一片黄叶。
出于美观整齐的考虑,小扣想要将收藏集中树叶的排列调整成「红、黄、红」三部分。每部分树叶数量可以不相等,但均需大于等于 1。每次调整操作,小扣可以将一片红叶替换成黄叶或者将一片黄叶替换成红叶。请问小扣最少需要多少次调整操作才能将秋叶收藏集调整完毕。
示例 1:
输入:leaves = "rrryyyrryyyrr"
输出:2
解释:调整两次,将中间的两片红叶替换成黄叶,得到 "rrryyyyyyyyrr"
示例 2:
输入:leaves = "ryr"
输出:0
解释:已符合要求,不需要额外操作
提示:
3 <= leaves.length <= 10^5
leaves 中只包含字符 'r' 和字符 'y'
思路:我们假设中间黄叶的区间为(j,i],则整个序列分成了三部分:(0,j]、(j,i]和(i,n]。
对于第一个区间,需要操作的次数是黄叶的数量,我们定义前缀和sum[i]表示区间(0,i]内黄叶的数量,则第一个区间的操作次数为sum[j],同理第三个区间为sum[n]-sum[i],而第二个区间的操作次数是红叶的数量,为i-j-(sum[i]-sum[j])。我们将三个式子合并到一起并合并同类项得总的操作次数为:sum[n]+i-2*sum[i]-(j-2*sum[j])。而为了让答案尽可能大,我们的减数项j-2*sum[j]一定要尽可能大,因此我们找到了线性解决该问题的方法,详见以下代码:
class Solution {
public int minimumOperations(String leaves) {
int n = leaves.length(), ans = n;
int[] sum = new int[n + 1];
for (int i = 1; i <= n; i++)
sum[i] = sum[i - 1] + (leaves.charAt(i - 1) == 'y' ? 1 : 0);
for (int i = 1; i < n; i++)
sum[i] = i - 2 * sum[i];
for (int i = 2, j = sum[1]; i < n; i++) {
ans = Math.min(ans, sum[n] + sum[i] - j);
j = Math.max(j, sum[i]);
}
return ans;
}
}