JAVA程序设计：秋叶收藏集（LCP 19）

小扣出去秋游，途中收集了一些红叶和黄叶，他利用这些叶子初步整理了一份秋叶收藏集 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;

    }

}