(leetcode每日打卡)秋叶收藏集【动态规划】
(leetcode每日打卡)秋叶收藏集【动态规划】
LCP 19.秋叶收藏集
算法
动态规划
时间复杂度O(n)
1.题目要求最终形成[红、黄、红]三部分，每部分数量可以不相等，问最终调整操作数量最小是多少。这道题一开始考虑暴力去做，枚举两个分界点，即红黄，黄红之间的分界点的位置，但由于长度是1e5，时间复杂度为O(n^2)级别，故此法作废。
2.通过查看官方题解，了解到这道题可以使用动态规划去做，可以将时间复杂度优化到O(n)级别，为方便查阅复习，现结合自己的理解写下该题解。具体如下：
可以定义一个数组f[i][j]，表示符合要求的最小的操作数，即使leaves[]数组从0到i的值符合题目规范的最小的操作数。i的范围是[0,leaves.length)，j的范围是[0,2]，其中0表示当前叶子为红色(在黄色前面)，1表示当前叶子为黄色，2表示当前叶子为红色(在黄色后面)。
初始时f[0][0]的值会由第一片叶子的颜色决定。下面分情况讨论j的不同值的情况：
(1)当j=0时，f[i][0] = f[i-1][0] + isYellow(i)。isYellow函数会根据叶子的颜色返回对应的布尔值；同时需注意j=0这种情况下i最大为leaves.length-3，因为题目要求每部分叶子数量至少为1个；
(2)当j=1时，f[i][1] = min(f[i-1][0],f[i-1][1]) + isRed(i) 。该叶子左面的叶子的颜色可能是红色，也可能是黄色，取形成前面两种情况操作的最小值即可。同时需要判断当前叶子是否黄色。在这种情况下i最大值为leaves.length-2，最小值为1；
(3)当j=2时,f[i][2] = min(f[i-1][1],f[i-1][2]) + isYellow(i);。该叶子左面的叶子的颜色可能是红色，也可能是黄色，取形成前面两种情况操作的最小值即可。同时需要判断当前叶子是否为红色。在这种情况下i最大值为leaves.length-1，最小值为2；
最终返回结果为f[leaves.length - 1]。
C++代码
//一开始是用函数调用的，但超时，故直接判断
class Solution {
public:
/*
bool isYellow(string leaves, int u){
return (leaves[u] == 'y');
}
bool isRed(string leaves, int u){
return (leaves[u] == 'r');
}
*/
int minimumOperations(string leaves) {
int len = leaves.length();
vector<vector<int> > f(len, vector<int>(3, INT_MAX));
//f[0][0] = isYellow(leaves, 0);
f[0][0] = (leaves[0] == 'y');
for(int i = 1; i < len; i++){
int yellow = (leaves[i] == 'y');
int red = (leaves[i] == 'r');
if(i < len - 2){
//f[i][0] = f[i-1][0] + (int)isYellow(leaves, i);
f[i][0] = f[i-1][0] + yellow;
}
if(i >= 1 && i < len - 1){
//f[i][1] = min(f[i-1][0],f[i-1][1]) + (int)isRed(leaves, i);
f[i][1] = min(f[i-1][0],f[i-1][1]) + red;
}
if(i >= 2){
//f[i][2] = min(f[i-1][1],f[i-1][2]) + (int)isYellow(leaves, i);
f[i][2] = min(f[i-1][1],f[i-1][2]) + yellow;
}
}
return f[len - 1][2];
}
};