校招真题练习032 连续相同字符串(头条)

连续相同字符串(编程题2)

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

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

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

 

def change(string,target,m,n):
    left = 0
    changelist = []
    changecount = 0
    maxlen = 0
    i,j = 0,0
    while i < n:
        if string[i] == target:
            i += 1
        else:
            if changecount < m:
                changecount += 1
                changelist.append(i)
                i += 1
            else:
                curleng = i - left
                if curleng > maxlen:
                    maxlen = curleng
                left = changelist[j] + 1
                j += 1
                changecount -= 1
    if len(changelist) > j:
        curleng = n - left
        if curleng > maxlen:
            maxlen = curleng
    return maxlen

def main():
    ary1 = list(map(int,input().split()))
    n,m = ary1[0],ary1[1]
    string = input().strip()
    c1 = change(string,'b',m,n)
    c2 = change(string,'a',m,n)
    if c1 >= c2:
        print(c1)
    else:
        print(c2)

if __name__ == '__main__':
    main()

 

算法思路：双指针，滑动窗口。

依据题意，需要尝试两种转换方式：1.将'a'转换为'b'，2.将'b'转换为'a'。

对于每一种转换，滑动窗口内部是完成m次转换的区域。当完成m次转换之后，计算当前区域的长度，并进行窗口的滑动。