LeetCode LCP19 秋叶收藏集

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

首先求得y和r的前缀和。

在字符串的过程中有两个指针，一个指向y的开头，另一个指向y的结尾。如果两个指针全部遍历的话时间复杂度为平方复杂度，超时。这道题的关键在于遍历一个指针，而且同时保留一个最小项，利用这种方法求得结果。

官方题解

class Solution:
    def minimumOperations(self, leaves: str) -> int:
        cnt_r,cnt_y=0,0
        nums_r=[]
        nums_y=[]
        for c in leaves:
            if c=='r':
                cnt_r+=1
            nums_r.append(cnt_r)
            if c=='y':
                cnt_y+=1
            nums_y.append(cnt_y)
        min_change=float('inf')
        sub=nums_y[0]-nums_r[0]
        for y in range(1,len(leaves)-1):
            min_change=min(min_change,nums_r[y]-nums_y[y]+cnt_y+sub)
            sub=min(sub,nums_y[y]-nums_r[y])
        return min_change