原题地址926. 将字符串翻转到单调递增
如果一个由 '0'
和 '1'
组成的字符串,是以一些 '0'
(可能没有 '0'
)后面跟着一些 '1'
(也可能没有 '1'
)的形式组成的,那么该字符串是单调递增的。
我们给出一个由字符 '0'
和 '1'
组成的字符串 S
,我们可以将任何 '0'
翻转为 '1'
或者将 '1'
翻转为 '0'
。
返回使 S
单调递增的最小翻转次数。
示例 1:
输入:"00110"
输出:1
解释:我们翻转最后一位得到 00111.
示例 2:
输入:"010110"
输出:2
解释:我们翻转得到 011111,或者是 000111。
示例 3:
输入:"00011000"
输出:2
解释:我们翻转得到 00000000。
提示:
1 <= S.length <= 20000
S 中只包含字符 '0' 和 '1'
算法
先把字符串转换成数组形式cal记录其连续的0或1的个数。例如"0001100"转换成[3,2,2,0];"1110"转换成[0,3,1,0]。
因此cal中偶数位对应"0",奇数位对应"1"。让最终的字符串递增,就是找到一个位置i,使得之前的都为"0",之后的都为"1"。
也就是把i之前的奇数位全翻转,之后的偶数位全翻转。
代码
class Solution:
def minFlipsMonoIncr(self, S):
"""
:type S: str
:rtype: int
"""
tmp, tmpcal = 0, 0
cal = []
for s in S:
if s == str(tmp):
tmpcal += 1
else:
tmp = 1 - tmp
cal.append(tmpcal)
tmpcal = 1
cal.append(tmpcal)
if tmp == 0:
cal.append(0)
to0, to1 = 0, 0
for i in range(0, len(cal), 2):
to1 += cal[i]
ans = to0 + to1
for i in range(0, len(cal), 2):
ans = min(ans, to0 + to1 - cal[i])
to0 += cal[i + 1]
to1 -= cal[i]
ans = min(ans, to0 + to1)
return ans