LCP 19. 秋叶收藏集(C++)---动态规划解题

题目详情

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

示例 1：

输入：leaves = "rrryyyrryyyrr"

输出：2

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

示例 2：

输入：leaves = "ryr"

输出：0

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

提示：

• 3 <= leaves.length <= 10^5
• leaves 中只包含字符 'r' 和字符 'y'

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/UlBDOe
 

——题目难度:中等


 

---

-使用三个数组解题的代码（带注释）

class Solution {
public:
    int minimumOperations(string leaves) {
    	if (leaves.size() == 0) return 0;
		vector<vector<int>> dp(3, vector<int>(leaves.size(), 0));
		//dp[0][i] 表示 从开头到下标 i 全部修改为「红色」 需要修改的最小次数
		//dp[1][i] 表示 从开头到下标 i 修改为「红、黄」需要修改的最小次数
		//dp[2][i] 表示 从开头到下标 i 修改为「红、黄、红」需要修改的最小次数 
		for(int i = 0; i < leaves.size(); i++) {
			if (i == 0) {
				dp[0][i] = (leaves[i] != 'r');
			} else {
				dp[0][i] = dp[0][i - 1] + (leaves[i] != 'r');
			}
			
			if (i == 1) {
				dp[1][i] = dp[0][i - 1] + (leaves[i] != 'y');
			} else if (i > 1) {
				dp[1][i] = min(dp[0][i - 1] + (leaves[i] != 'y'), 
				dp[1][i - 1] + (leaves[i] != 'y'));
			}
			
			if (i == 2) {
				dp[2][i] = dp[1][i - 1] + (leaves[i] != 'r');
			} else if (i > 2) {
				dp[2][i] = min(dp[1][i - 1] + (leaves[i] != 'r'),
				dp[2][i - 1] + (leaves[i] != 'r'));
			}
		}
		
		return dp[2].back();
    }
};