题目
小扣出去秋游,途中收集了一些红叶和黄叶,他利用这些叶子初步整理了一份秋叶收藏集 leaves, 字符串 leaves 仅包含小写字符 r 和 y, 其中字符 r 表示一片红叶,字符 y 表示一片黄叶。
出于美观整齐的考虑,小扣想要将收藏集中树叶的排列调整成「红、黄、红」三部分。每部分树叶数量可以不相等,但均需大于等于 1。每次调整操作,小扣可以将一片红叶替换成黄叶或者将一片黄叶替换成红叶。请问小扣最少需要多少次调整操作才能将秋叶收藏集调整完毕。
示例一
输入:leaves = “rrryyyrryyyrr”
输出:2
解释:调整两次,将中间的两片红叶替换成黄叶,得到 “rrryyyyyyyyrr”
思路分析
- 首先我们必须保证两头都是r叶子。
- 接下来处理中间的叶子。中间的叶子需要保证是y的叶子。但是因为可能出现y和r的交叉,我们此时可以选择将y变成r或者r变成y。因此一次变换之后我们可以保证当前的字符串是连续的,例如遇到ryyyryr,中间是yyyry我们对于yr来说,选择将r变成y就可以和前面的两个yy相连,这样我们能保证当前字符串是连续的。那么我们其实是可以直接删除yr这个无效字符串的。因为此时前面相当于还有y在撑着。但是如果对于ryrrrryyr来说,消除yr的操作相当于是将y变成r,此时所有的r都能相连。
- 为了维护我们的消除次数,其实我们是需要当前来维护一个长度值tot,这个数值保存目前可以在消除的操作情况下,连续的y的最长长度。对于上面第一个例子来说,长度是3,对于第二个例子来说,长度是2.
- cnt就是模拟这个消除过程,遇到y,+1;遇到r,如果前面还有y匹配的情况下,相当于同时消除yr,-1。
代码
class Solution {
public:
// ryr
int minimumOperations(string leaves) {
int n = leaves.size(), cnt = 0, ans = 0;
if(leaves[0] == 'y') ans ++;
if(leaves[n - 1] == 'y') ans++;
for(int i = 1;i < n - 1;i++){
if(leaves[i] == 'y') cnt++;
}
if(cnt == 0) return ans + 1;//表示当前中间所有的字符都是r,此时随便替换一个。
//如果中间是y和r的交差。一个y和r连着的时候,我们选择y-r或者r-y,这种方式就相当于做了一次变换把中间这两个字符给消除了。
int tot = 0, cur = 0;
for(int i = 1;i < n - 1;i++){
if(leaves[i] == 'y') cur++;
else if(cur > 0) cur--;
tot = max(tot, cur);//保存连续y的最长长度
}
return cnt - tot + ans;
}
};