LCP 19. 秋叶收藏集

小扣出去秋游，途中收集了一些红叶和黄叶，他利用这些叶子初步整理了一份秋叶收藏集 leaves， 字符串 leaves 仅包含小写字符 r 和 y， 其中字符 r 表示一片红叶，字符 y 表示一片黄叶。
出于美观整齐的考虑，小扣想要将收藏集中树叶的排列调整成「红、黄、红」三部分。每部分树叶数量可以不相等，但均需大于等于 1。每次调整操作，小扣可以将一片红叶替换成黄叶或者将一片黄叶替换成红叶。请问小扣最少需要多少次调整操作才能将秋叶收藏集调整完毕。

示例 1：

输入：leaves = "rrryyyrryyyrr"

输出：2

解释：调整两次，将中间的两片红叶替换成黄叶，得到 "rrryyyyyyyyrr"

示例 2：

输入：leaves = "ryr"

输出：0

解释：已符合要求，不需要额外操作

提示：

1. 3 <= leaves.length <= 10^5
2. leaves 中只包含字符 ‘r’ 和字符 ‘y’
3. 通过次数3,420提交次数11,109

解答

动态规划，由于最终要实现rrr...yyy...rrr这样的排列，因此当遍历到字符串第i位时，总共有可能存在三种模式：rrr...，rrr...yyy...，以及rrr...yyy...rrr...，因此分别用dp数组记录到位置i实处现三种模式所需要更改的最少字符数，注意开头和结尾必须都是字符r。

class Solution {
public:
    int minimumOperations(string leaves) {
        int len = leaves.length();
        vector<vector<int>> dp(3, vector<int>(len, 1e5));
        int isYellow = leaves[0]=='y' ? 1 : 0;
        dp[0][0] = isYellow;
        for(int i=1;i<len;i++){
            isYellow = leaves[i]=='y' ? 1 : 0;
            dp[0][i] = dp[0][i-1] + isYellow;
            // 有可能是`rrr...`模式或者`rrr...yyy...`模式
            dp[1][i] = min(dp[0][i-1], dp[1][i-1]) + (isYellow ? 0 : 1);
            // 有可能是`rrr...yyy...`模式或者`rrr...yyy...yyy...`模式
            dp[2][i] = min(dp[2][i-1], dp[1][i-1]) + isYellow;
        }
        return dp[2][len-1];
    }
};