字节跳动校招----编程题2（Python）

一、题目

题目描述

有一个仅包含’a’和’b’两种字符的字符串s，长度为n，每次操作可以把一个字符做一次转换（把一个’a’设置为’b’，或者把一个’b’置成’a’)；但是操作的次数有上限m，问在有限的操作数范围内，能够得到最大连续的相同字符的子串的长度是多少。

输入描述:

第一行两个整数 n , m (1<=m<=n<=50000)，第二行为长度为n且只包含’a’和’b’的字符串s。

输出描述:

输出在操作次数不超过 m 的情况下，能够得到的 最大连续 全’a’子串或全’b’子串的长度。

示例1

输入

8 1
aabaabaa

输出

5

说明

把第一个 'b' 或者第二个 'b' 置成 'a'，可得到长度为 5 的全 'a' 子串。

二、分析及代码

题意：给一串由a和b组成的字符串，可以变换m个字母，求变化后连续相同字符的最大长度

分析：将字符a和字符b的所在的元索引分别提取出来，max=max(p_a[i+m]-p_a[i-1]-1)。本子上画画即可明白这个道理。代码如下：

[n, m] = [int(x) for x in input().split(' ')]
s = input()
a=s.count('a')
b=s.count('b')
res = 0
p_a = [-1]    # 添加-1是为了计算真实的第一个
p_b = [-1]

for i in range(n):    # 找a和b的索引
    if s[i]=='a':
        p_a.append(i)
    else:
        p_b.append(i)
if m>=a or m>=b:
    res = n
else:
    res=0
    for i in range(1,len(p_a)):
        if i+m<len(p_a):
            maxlen = p_a[i+m]-p_a[i-1]-1
            if maxlen>res:
                res=maxlen
        else:
            break
        
    for i in range(1,len(p_b)):
        if i+m<len(p_b):
            maxlen = p_b[i+m]-p_b[i-1]-1
            if maxlen>res:
                res=maxlen
        else:
            break
            
print(res)