题目:
小扣出去秋游,途中收集了一些红叶和黄叶,他利用这些叶子初步整理了一份秋叶收藏集 leaves, 字符串 leaves 仅包含小写字符 r 和 y, 其中字符 r 表示一片红叶,字符 y 表示一片黄叶。
出于美观整齐的考虑,小扣想要将收藏集中树叶的排列调整成「红、黄、红」三部分。每部分树叶数量可以不相等,但均需大于等于 1。每次调整操作,小扣可以将一片红叶替换成黄叶或者将一片黄叶替换成红叶。请问小扣最少需要多少次调整操作才能将秋叶收藏集调整完毕。
思路:参考官方解答
将序列视为3个状态块,状态0表示最前面的红色部分,状态1表示中间的黄色部分,状态2表示最后面的红色部分。
f[i][j]:表示 对于第 0 片叶子到第 i片叶子(记为 leaves[0…i])进行调整操作,并且第 i 片叶子处于状态 j 时的最小操作次数。
说明:若第i片叶子处于状态0,那么第i-1片叶子只能处于状态0;
若第i片叶子处于状态1,那么第i-1片叶子可处于状态0,1;若第i片叶子处于状态2,那么第i-1片叶子可处于状态1,2。因为每个状态至少要有1片叶子,故第i片叶子处于状态2时,第i-1片叶子不可能处于状态0。
初始状态:f[0][0]=(leaves[0]==‘y’)
特殊状态:f[0][1]=f[0][2]=f[1][2]=float(inf)
(特殊状态表示该状态不存在,故定义为无穷大)
最终结果:f[n-1][2]
(叶子是从0开始计数的,故最终结果取f[n-1][2])
解答:
class Solution:
def minimumOperations(self, leaves: str) -> int:
n=len(leaves)
f=[[0,0,0] for _ in range(n)]
f[0][0]=(leaves[0]=='y')
f[0][1]=f[0][2]=f[1][2]=float(inf)
for i in range(1,n):
isYellow=(leaves[i]=='y')
isRed=(leaves[i]=='r')
f[i][0]=f[i-1][0]+isYellow
f[i][1]=min(f[i-1][0],f[i-1][1])+isRed
if(i>=2):
f[i][2]=min(f[i-1][1],f[i-1][2])+isYellow
return f[n-1][2]