今日头条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' 子串。我的方法：

#include<stdio.h>
#include<iostream>
#include<string>
#include<algorithm>

using namespace std;

int main() {
	int n, m;
	int j = 0, k = 0;
	int maxLen = 0;
	string s;
	cin >> n >> m;
	cin >> s;
	int astring[50000] = { 0 };
	int bstring[50000] = { 0 };
	for (int i = 0; i<n; i++) {
		if (s[i] == 'a')
			astring[j++] = i;
		else
			bstring[k++] = i;
	}
	if (m >= j || m >= k) {
		cout << s.size() << endl;
		return 0;
	}
	if (m <= 0) {
		int j = 0;
		for (int i = 0; i < s.size(); i++) {
			maxLen = max(maxLen, i - j);
			while (s[j] != s[i])
				j++;
		}
		cout << maxLen << endl;
		return 0;
	}
	//首位处理，0<m<j,0<m<k
	int maxLen1 = max(astring[m], int(s.size() - astring[j - m-1]-1));   
	for (int i = 1; (i + m) < j; i++) {
		maxLen1 = max(astring[i + m] - astring[i - 1] - 1, maxLen1);
	}
	int maxLen2 = max(bstring[m], int(s.size() - bstring[k - m-1]-1));
	for (int i = 1; (i + m) < k; i++) {
		maxLen2 = max(bstring[i + m] - bstring[i - 1] - 1, maxLen2);
	}
	maxLen = max(maxLen1, maxLen2);
	cout << maxLen << endl;
	return 0;
}

方法二：

#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;