【牛客网】2018字节跳动编程2——最大连续相同字符子串（自己起的名字）

链接：https://www.nowcoder.com/questionTerminal/dcc301bc11a7420b88afdbd272299809
来源：牛客网

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

解题思路

step1：遍历字符串s，分别把a的下标和b的下标分别存进两个数组va，vb。

step2：分为a替换b获取最大子串和b替换a。

a替换b时，遍历vb = [num1,num2, , , ,]；

取包含m个b的最大区间，遍历num1的时候比较特殊，获得的子串长度就是vb[2]

从vb[2]开始遍历获取的子串长度公式为vb[i+m]-vb[i-1]-1

b替换a时类似。

代码

#include <iostream>
#include <vector>
#include <string>
#include <algorithm>
using namespace std;
int main()
{
	int n, m, pre = 0, cur = 0, numa = 0, numb = 0;
	cin >> n >> m;
	char c;
	string s;
	int ii = n;
	while (ii--)
	{
		cin >> c;
		s += c;
	}
	vector<int> va, vb;//分别存a的下标，b的下标

	for (int i = 0; i < n; i++)
	{
		if(s[i] == 'a')
			va.push_back(i);
		if(s[i]== 'b')
			vb.push_back(i);
	}
	vector<int> res;

	//把m个b替换成a，获取最长串
	res.push_back(vb[0 + m]);//遍历第一个b的下标的时候比较特殊
	for (int i = 1; i + m < vb.size(); i++)
	{
		res.push_back(vb[i + m] - vb[i - 1] - 1);
	}

	//把m个a替换成b，获取最长串
	res.push_back(va[0 + m]);//遍历第一个a的下标的时候比较特殊
	for (int i = 1; i + m < va.size(); i++)
	{
		res.push_back(va[i + m] - va[i - 1] - 1);
	}

	sort(res.begin(), res.end());
	cout << res[res.size() - 1] << endl;

	system("pause");
	return 0;
}