题目描述:
小扣出去秋游,途中收集了一些红叶和黄叶,他利用这些叶子初步整理了一份秋叶收藏集 leaves
, 字符串 leaves
仅包含小写字符 r
和 y
, 其中字符 r
表示一片红叶,字符 y
表示一片黄叶。 出于美观整齐的考虑,小扣想要将收藏集中树叶的排列调整成「红、黄、红」三部分。每部分树叶数量可以不相等,但均需大于等于 1。每次调整操作,小扣可以将一片红叶替换成黄叶或者将一片黄叶替换成红叶。请问小扣最少需要多少次调整操作才能将秋叶收藏集调整完毕。
示例:
输入:leaves = "rrryyyrryyyrr"
输出:2
解释:调整两次,将中间的两片红叶替换成黄叶,得到 "rrryyyyyyyyrr"
题解:
这道题其实很符合动态规划的特征,即后续调整需要根据前序状态来决定,但是不同的是每个位置的树叶其实是有三个状态的:1.左边的红色 2.中间的黄色 3.右边的红色。不妨建立二维数组dp[i][j],其中第一维的i代表了到第i个叶子需要的最小的调整次数,第二维j代表了i树叶的状态。对于状态1来说即dp[i][0](0代表状态1,后续以此类推),dp[i][0] = dp[i-1][0] + isyellow;即如果第i个叶子置为状态1那么其前面叶子的状态也只能为1,同时判断需不需要将i叶子本身进行换色。dp[i][1] = min(dp[i-1][0], dp[i-1][1]) + isred。对于第i个叶子置为状态2的情况,它的前序既可以是状态1,也可以是状态2,同时考虑是否本身需要换色。同理,dp[i][2] = min(dp[i-1][1], dp[i-1][2]) + isyellow。对于第i个叶子是状态3的情况,它的前序叶子状态只能是2或3,同时 同时考虑是否本身需要换色。最后,由于状态的数量不能大于叶子的数量,所以dp[0][1] = dp[0][2] = dp[1][2] = INT_MAX;边界条件为:第0个叶子必须为红色。
代码如下:
class Solution {
public:
int minimumOperations(string leaves) {
if(leaves.empty()){
return 0;
}
int n = leaves.length();
vector<vector<int>> dp(n,vector<int>(3));
dp[0][0] = leaves[0] == 'r' ? 0 : 1;
dp[0][1] = dp[0][2] = dp[1][2] = INT_MAX;
for(int i = 1; i < n; i++){
cout<<i<<endl;
int isyellow = static_cast<int>(leaves[i] == 'y');
int isred = static_cast<int>(leaves[i] == 'r');
dp[i][0] = dp[i-1][0] + isyellow;
dp[i][1] = min(dp[i-1][0], dp[i-1][1]) + isred;
if(i >= 2){
dp[i][2] = min(dp[i-1][1], dp[i-1][2]) + isyellow;
}
}
return dp[n-1][2];
}
};