编程题2-最大连续的相同字符(2018字节跳动校招)

题目描述

有一个仅包含’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' 子串。

算法分析

以b替换a为例，记录每一个b的位置，那么将连续m个b的置换为a必然存在最长的字串。

对索引为i的b而言，将前面m个b置换为a，此时索引为i的b与索引为i-m-1的b之间的字串的长度为loc[i] - loc[i-m-1]-1。

初始化时，i从m+1开始，因为前面不足m个b的时候，只要全部置为a即可。

提交代码：

#include<iostream>
#include<vector>
#include<algorithm>
#include<string>

using namespace std;

string str;
int n, m;

int change_alpha(char k)
{
    vector<int> loc;
    for (int i = 0; i < n; ++i)
        if (str[i] == k) loc.push_back(i);
    loc.push_back(n);
    int max_length = loc[m];
    
    for (int i = m + 1; i < loc.size(); ++i) {
        // loc[i]当前k实际索引, loc[i-m-1] i的前面m个数翻转之后，再前面的未翻转k的索引
        // loc[i] - loc[i-m-1] - 1 之间翻转m个k之后，中间字符的长度。
        max_length = max(max_length, loc[i] - loc[i-m-1] - 1);
    }
    return max_length;
}

int main()
{
    cin >> n >> m;
    cin >> str;
    
    cout << max(change_alpha('a'), change_alpha('b')) << endl;
}